Modified UML diagram