Added prints to visualizer.py