Survival analysis in Python B)