Added the right UML-diagram