Fixed some gravity problems