Pretty Big Update

Made some updates to the visualizer. Added the compatibility to be able
to show merged trajectories as policies. Plus some clean up.