Fix search ux and bugs.