Added more control to the gravity