minor changes + diagrams