Added an image and minor code cleanup.