Added the search functionality