Set up the button color changing