Create two_sum.py