started implementing shallow_water