Slight modification in Part1 max_2_sum