Search functionality update