minor changes in diagramm