Update two_body_interaction.py