Add python script to plot the trajectory of the particles