UML diagram updated