updated gen_graph.py