Fixed the N-body part after the refactoring