Create generate_graph.py