Fixed the gravity demo (distance update thingy)