Adding classes back to image.