Made some changes to puppet script