added caption and comments