Add search functionality to UI