Small fixes for nbody