Update n_body_brute_force.py