Add class to button