Update graph_api.py