Added visualizer code