Extension of artificial viscosity technique for solving 2D non-hydrostatic shallow water equations