Clean up Visualizer

Made the Visualizer functional.  Some minor code fixes.