Adding dedicated class for button