It seems you only have x and y vectors in there and no z vector which is what gravity relies on.(not familiar with the method)
Then again, you have a double vector statement rather than one.

Never liked that sort of gravity control as it takes too much to balance forces I find.

Try vector(x_momentum, 0,z_momentum)