corrected UML graph