Update generate_graph.py