Thank you for the comment and feedback!
I will have to test around with the momentum in air vs on ground issue. From what I understand, the momentum retention should be the same in both cases. However, I totally agree with your comments. As we develop the game further, I think there should be more momentum retention (less friction) in the air as opposed to on the ground. It definitely makes more intuitive sense.