Added color change to the button