Two sum problem python solution