Added some search functionality to UI