Add conditional cohort colors