Update build_graph.py