Create two-sum.py