Redo som UML diagram for second part