updated plot_waterfall