edited color_fun