Add the search function