added more classes to UML