If we assume the divergent part of the eddy PV flux on isopycnal surfaces is directed down the mean PV gradient, the resulting expression for the flux in height coordinate is made of a relative vorticity, beta, and stretching contributions. We implement and test our scheme in a coarse resolution configuration of the MIT ocean circulation model using an idealized sector-channel geometry with a flat bottom and one active tracer. The model is forced both mechanically and thermodynamically. We conduct a series of experiments in which the sensitivity of the solution to the formulation of the PV flux ``force'' is studied. We also experiment with different choices of boundary conditions for the PV flux. Finally, the scheme is implemented in a global version of the MIT model (at a horizontal resolution of 2.8 degrees) forced by realistic wind-stress and heat and freshwater fluxes.