Add focus state to the search box and visual tweak