Updated puppet provision code