improved diagramm