Add button click listener