Update make_design_graph.py