add code to show image and classes