UI improvements to search