Add search method with passing tests