Small improvements to diagrams