Add badge count for search results