Added handler for button.