diff --git a/.gitignore b/.gitignore
index ab9f877..5fec197 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,4 +10,7 @@ data/
 .tools/
 
 dbschema/migrations/
-*.jsonl
\ No newline at end of file
+
+*.jsonl
+dataset/
+models/
diff --git a/justfile b/justfile
index bf493c0..25ec178 100644
--- a/justfile
+++ b/justfile
@@ -171,7 +171,7 @@ curate-dataset input="good_moves-*.jsonl" output="data/dataset/best_moves.jsonl"
 analyze-dataset input="good_moves-*.jsonl" output="":
   if [ -n "{{output}}" ]; then python -m server.DatasetStats --input "{{input}}" --output "{{output}}"; else python -m server.DatasetStats --input "{{input}}"; fi
 
-train-ai input="data/dataset/best_moves.jsonl" rl_input="data/dataset/rl_bootstrap.jsonl" output="models/battlesnake_softmax_v2.json" eval_split="0.2" seed="42" epochs="14" lr="0.08":
+train-ai input="dataset/best_moves.jsonl" rl_input="dataset/rl_bootstrap.jsonl" output="models/battlesnake_softmax_v2.json" eval_split="0.2" seed="42" epochs="14" lr="0.08":
   if [ -f "{{rl_input}}" ]; then python -m server.TrainBattleSnakeAI --input "{{input}}" --input "{{rl_input}}" --output "{{output}}" --eval-split "{{eval_split}}" --seed "{{seed}}" --epochs "{{epochs}}" --lr "{{lr}}"; else python -m server.TrainBattleSnakeAI --input "{{input}}" --output "{{output}}" --eval-split "{{eval_split}}" --seed "{{seed}}" --epochs "{{epochs}}" --lr "{{lr}}"; fi
 
 run-trained model="models/battlesnake_softmax_v2.json" port="8000":