further work on step function in shallow_water