Added a button class