Add search code