Add handler for button press.