Add puppet highlighting