Survival analysis in Python ;)