Add UML for iteration 3