updated cohort schema