add option for color