Create viz.py