updated viz.py
updated viz.py