ENH: Add class to plot 3D trajectories