more work on the buttons