working 2d shallow water equations