Updated UML documentation