Moving to python-graph