Fix search box functionality