adding in survival graph