revise code for TwoSum