Updated highlighting test