Add UML diagrams to documentation