updated puppet script