X-Git-Url: https://git.sommitrealweird.co.uk/advent-of-code-2021.git/blobdiff_plain/93dfe872eef986673cae31e8e985c547a09209f7..39130f89f58490298b833e8f9e0036a4844f4a3e:/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