made some changes in diagramm