Adding survival analysis