Working on cohort fine-tuning