Made minor changes to the button code.