Added more classes to the class diagram