PDE solver for 1 layer shallow water model