Changed structure of init.pp to use Case statement