Updated the UML Diagrams