Add search logic