Add the attribute class for highlighting