Gravity

Code review for the gravity implementation