add class for button