Made gravity work more properly