Add state diagramm