Add the UML diagrams