Changes in class diagramm