additional figure and improved captioning of images