Updated parse_query hook to check for search as well.