Added UML class diagram