More fixes in visualizer.py'
Add visualization script