add class for image