UI Improvements for search