Updated some of the UML-diagrams