Update graph.py