Fixed up how gravity works