Add the missing 'highlight' class