Create visualize.py