Add a class for the button