Python Plot 2d Trajectory