remove pl.figure in plot_surv_2d