Added functionality that appends image to it's corresponding div. This allows the divs class to encompass the image.  Added functionality data-class attribute that is applied to the image.