Working to UML Class Diagram with some of the main classes