Added visualizer back in