Add custom highlighting