Integrate event handling for button