improve a couple of diagrams