Update visualizer.py