Now returns correct answer
Removed graph.