Changed Module to Class so that it works with hiera