updated UML model