WIP color function