Add puppet support for highlighting