working on cohort - survival analysis