Made some minor changes to the visualizer.