Added the modified UML