Updated trajectory_animation.py