added Sean's work on updating the graph class