more changes to visualizer