Adding class to buton