Add more documentation to UML diagrams