X-Git-Url: https://git.sommitrealweird.co.uk/advent-of-code-2021.git/blobdiff_plain/64d8769ea65405cdc0a1f8e65a8a1fa25bbf71df..9211cd2ef3bbf35f1328c86c5c6290005c014cb3:/day01/count.sh

diff --git a/day01/count.sh b/day01/count.sh
index 990433d..86a89b1 100755
--- a/day01/count.sh
+++ b/day01/count.sh
@@ -9,7 +9,9 @@ cur_window=1
 
 declare -a windows
 
-for num in $(<input.txt); do
+filename="${1:-input.txt}"
+
+for num in $(<"$filename"); do
     cur_line=$((cur_line+1))
     windows[$cur_window]=$num
     if [ $cur_window -gt 1 ]; then