Add functionality to button