made the button class work!