Updated init.pp to add support for Puppet Enterprise