Add the function of the search button