vit function

Visualize individual trajectories