Add highlighting for company-mode