Updating puppet code