edited the graph.py script