edited the UML Diagrams