The class needs to be on the button