Add option to specify custom colors