Update class diagramm a little