Search box ui improvements