getting the button to work a bit better