Updated plots and script for survival analysis