tweaked the button class