Update Python wrapper for refactored Trajectory