more work on the button