Add captions to diagrams