Cleaning puppet code