correct diagramm
add class