Updated the styling and generation of captions