Added more to DisplayGraph, added functions to other classes