Diagrammen klara

Lite småfix i koden också.