A cleaner alternative way to do three sum