You can actually set the class directly on the img tag, no need to wrap it in another div