Minor update to UML-diagram
Updated class diagram again