add draw_graph.py