Added classes to the class diagramm