Create generate_viz.py