Set a click listener of a button