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