Returning figure with multiple axes objects causing problem with Matplotlib animation via celluloid

Question

I am trying to return multiple axes objects on a single plot using the python function. The function takes in two parameters a and b and returns hi