added visualization comments and captions
added captions and descriptions