Got the classes button working with an image