UML added for iteration 3