Puts the button into a class