Fixing survival.py