Update generate_graphs.py