Oh, I forgot to mention in the game, that when you die in water you need to press Esc and then there's a respawn button (ups :) ), and the momentum just stopping is a bug, pretty annoying, didn't really know how to fix it, so I just leaved it there.
I am really happy that you liked it and thanks a lot for the feedback, it really helps me!! :D