rearrange imports in analyze_cohort