Got the buttons working