Added ability to visualize test trajectories