UML class diagram updated