added gravity
improve gravity