Add test for cohort-create