Fixed up puppet script a bit.