Better gravity z movement calculation