added ability to set color