UML Diagram Updated