Saving trajectories now work. GPU and CPU trajectories currently diverge.