Update documentation for UML diagrams