change div of image to a class