Updated UI for search