improve some diagrams