Add search box functionality