Updated visualizer.py