HW5 - getting rid of the graph