Update N bodies (python)