4Sum python