What if I wanted to give you the figure to do the animation on? I think ```plt.gcf``` would be better here. even nicer might be to accept figure as a kwarg.