Simplify color-for function