diff --git a/download.sh b/download.sh
index fd56b59..3054b45 100755
--- a/download.sh
+++ b/download.sh
@@ -1,8 +1,11 @@
 dataset_operation() {
     cat all_games.csv | shuf > all_games.csv.shuf
-    head -n 1880 all_games.csv.shuf > all_games.csv.test
-    head -n 3760 all_games.csv.shuf | tail -n 1880 > all_games.csv.dev
-    tail -n +3761 all_games.csv.shuf > all_games.csv.train
+    len1=$(cat ./all_games.csv.shuf | wc -l)
+    len2=$(($len1/10))
+    len3=$(($len2*2))
+    head -n $len1 all_games.csv.shuf > all_games.csv.test
+    head -n $len2 all_games.csv.shuf | tail -n $len1 > all_games.csv.dev
+    tail -n +($len2 + 1) all_games.csv.shuf > all_games.csv.train
     rm all_games.csv.shuf
     wc -l all_games.csv.*
 }