Fixed some small bug in the graph