Adding additional captions to diagram