Update ps7_visualize.py