Dhm/bug cohort args

@dianakhuang please review