Where g is the gravitational acceleration.

and gravitational_term

and made some other minor changes.

and for axisymmetical problems.