Replace this with python-graph dependecy