improve diagrams