UML diagrams updated