Add UML diagrams for iteration 2