Add button class