diff --git a/config/hooks/installed/encryption-setup.sh b/config/hooks/installed/encryption-setup.sh
index ab97caa..7d29891 100755
--- a/config/hooks/installed/encryption-setup.sh
+++ b/config/hooks/installed/encryption-setup.sh
@@ -47,10 +47,12 @@ CRYPTSETUP=y
 EOF
 
 # Add cryptsetup and dm-crypt to initramfs modules
-echo "dm_crypt" >> /etc/initramfs-tools/modules
-echo "aes_xts" >> /etc/initramfs-tools/modules
-echo "xts" >> /etc/initramfs-tools/modules
-echo "sha512" >> /etc/initramfs-tools/modules
+{
+    echo "dm_crypt"
+    echo "aes_xts"
+    echo "xts"
+    echo "sha512"
+} >> /etc/initramfs-tools/modules
 
 # Configure kernel command line for encrypted root
 if [ -f /etc/default/grub ]; then
diff --git a/src/firewall-setup.sh b/src/firewall-setup.sh
index 86b1cff..5f9b4a8 100755
--- a/src/firewall-setup.sh
+++ b/src/firewall-setup.sh
@@ -71,7 +71,7 @@ apply_firewall() {
 # Main setup
 main() {
   echo "Setting up dynamic firewall..."
-  apply_firewall
+  apply_firewall "${1:-}"
   echo "Firewall setup completed."
 }
 
diff --git a/src/security-hardening.sh b/src/security-hardening.sh
index 4b4523e..f98a6b1 100755
--- a/src/security-hardening.sh
+++ b/src/security-hardening.sh
@@ -131,15 +131,16 @@ EOF
 }
 
 # Function to apply all security configurations
+# shellcheck disable=SC2120
 apply_security_hardening() {
   echo "Applying security hardening..."
 
-  create_wifi_blacklist
-  create_bluetooth_blacklist
-  configure_ssh
-  configure_password_policy
-  configure_system_limits
-  configure_audit_rules
+  create_wifi_blacklist "${1:-}"
+  create_bluetooth_blacklist "${2:-}"
+  configure_ssh "${3:-}"
+  configure_password_policy "${4:-}"
+  configure_system_limits "${5:-}"
+  configure_audit_rules "${6:-}"
 
   echo "Security hardening completed."
 }