added class diagramm