added plot_waterfall