Fix for missing color_code_for method