Robustified the survival script