Changes to set color