Buttons now changes colour back to what it should be.