Updated visualizer