Set up UMLGraph