After implement the search UI