modify waterfall demo