1 files changed, 8 insertions, 9 deletions
diff --git a/manifests/client.pp b/manifests/client.pp
index 5eed5fc..84dd7ab 100644
--- a/manifests/client.pp
+++ b/manifests/client.pp
@@ -1,23 +1,22 @@
 # manifests/client.pp
 
-class sshd::client {
+class sshd::client(
+  $shared_ip = 'no',
+  $ensure_version = 'installed',
+  $manage_shorewall = false
+) {
 
-  case $sshd_shared_ip {
-    '': { $sshd_shared_ip = "no" }
-  }
-
-  case $operatingsystem {
+  case $::operatingsystem {
     debian,ubuntu: { include sshd::client::debian }
     default: {
-      case $kernel {
+      case $::kernel {
         linux: { include sshd::client::linux }
         default: { include sshd::client::base }
       }
     }
   }
 
-  if $use_shorewall{
+  if $manage_shorewall{
     include shorewall::rules::out::ssh
   }
-
 }