Update 3sum.py