Improved puppet script

Signed-off-by: Vlad <vg@vladgh.com>