minor changes to diagrams