update code for search box