Working on shallow water model