Some changes I made to graph