-Fixed some bugs with the visualization
-Minor UI tweaks