Solved the problem with the graph