Updated Puppet code