Enhance: add caption