From 210a977cffc926bb8aa1a35aa8ae93ed98386d0c Mon Sep 17 00:00:00 2001
From: Zakary Timson <zaktimson@gmail.com>
Date: Fri, 26 Jul 2024 23:11:35 -0400
Subject: [PATCH] Update swap.sh

---
 swap.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/swap.sh b/swap.sh
index fc9d4cc..b702858 100644
--- a/swap.sh
+++ b/swap.sh
@@ -1,6 +1,9 @@
 #!/bin/bash
 
-dd if=/dev/zero of=/swapfile bs=1m count=1024
+SIZE="1024"
+if [ -n "$1" ]; then SIZE="$1"; fi
+
+dd if=/dev/zero of=/swapfile bs=1m count="$SIZE"
 chown 600 /swapfile
 mkswap /swapfile
 swapon /swapfile