Updated puppet script