modify to clean-up nbody example