Changes to visualizer