Add the ability to choose the button to respond to