Add trajectory simulation
The final version