Update two_sum.py