Add signal handling for the button