Python: cleanup animate.py