Add a force to cancel gravity (or a portion of it), and add a force that is inversely proportional/exponential to velocity Z (to cancel it).
Something like
