fit_models function

Fit Survival Models