Improve the UX for the search box