Edited the graph impl