modify twoBody.py