Update and refresh code.

This is to make all the code run on modern version of Ruby and Puppet.

Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>