Set up a button listener.