Improves the diagrams part.