Updated UML diagrams