Add the option to use code for choosing colors