Updated the graph class