Add div class for image