Add back search to UI