Improve search bar UI on mobile