modified code for cohort size increase