Add the button click listener