Added matplotlib_anim code