The class needs to be on the button here as well.