Changing the way the buttons work