Make it listen to the button