simplified code and deleted cohort 2