Switched vortex_rhs to fast implementation of gradient