Update nbody to use map_with_location