Add puppet code