Add the size as a class to the image