use class for button