diff --git a/bin/qcode.bash b/bin/qcode.bash
index decfaf6..061f785 100755
--- a/bin/qcode.bash
+++ b/bin/qcode.bash
@@ -5,14 +5,14 @@ UPCURSOR_CLEAR_ANSI="\033[1A\033[K"
 
 select_random_drill() {
   local drill
-  drill=$(jq 'keys[]' q-codes.json \
+  drill=$(jq 'keys[]' "$1" \
     | sort -R \
     | head -n 1 \
     | xargs -I "%" jq "\
         \"%\" as \$qcode \
         | .[\$qcode] as \$qobj \
         | {qcode: \$qcode, question: \$qobj.question, answer: \$qobj.answer}" \
-      q-codes.json
+      "$1"
   )
   echo "$drill"
 }
@@ -33,7 +33,7 @@ run_drill() {
 
 main() {
   local random_drill
-  random_drill="$(select_random_drill)"
+  random_drill="$(select_random_drill "$@")"
   run_drill "$random_drill"
 }
 
@@ -44,5 +44,5 @@ echo -e "To exit drills, use \e[33mctrl-c\e[0m."
 echo ""
 
 while true; do
-  main
+  main "$@"
 done;
diff --git a/q-codes.json b/bin/qcodes.json
similarity index 100%
rename from q-codes.json
rename to bin/qcodes.json
diff --git a/flake.nix b/flake.nix
index fdf9fe2..13eda45 100644
--- a/flake.nix
+++ b/flake.nix
@@ -28,7 +28,7 @@
               jq
               figlet
             ];
-            text = ./bin/qcode.bash;
+            text = "${./bin/qcode.bash} ${./bin/qcodes.json}";
           };
           devShells.default = pkgs.mkShell {
             nativeBuildInputs = [ self'.packages.default ];