Update trajectory.py