Accidentally overwrote github code with code from Puppet Forge.
Adding missing code back in.