Trajectory now saves the timestep as well. Added unittest for Trajectory.