Removing the redundant vector in cohort.h