Improved search UI