Can now plot trajectories. Still a bit buggy!