Add code to install puppet