Fixed color function/legend on reg plots