diff --git a/data/.gitignore b/data/.gitignore
new file mode 100644
index 0000000..a535cf3
--- /dev/null
+++ b/data/.gitignore
@@ -0,0 +1,4 @@
+*
+
+!datasets
+!.gitignore
diff --git a/datasets/.gitignore b/datasets/.gitignore
new file mode 100644
index 0000000..cec9082
--- /dev/null
+++ b/datasets/.gitignore
@@ -0,0 +1,3 @@
+*
+
+!.gitignore
diff --git a/default.env b/default.env
index 2db7f32..c9a94b1 100644
--- a/default.env
+++ b/default.env
@@ -1,2 +1,2 @@
 YOLO_WEB_URL=http://localhost:3333
-YOLOv5=../yolo_train/
\ No newline at end of file
+DATASETS./datasets
diff --git a/scripts/install_web.sh b/scripts/install_web.sh
index 33521ba..f646739 100755
--- a/scripts/install_web.sh
+++ b/scripts/install_web.sh
@@ -2,10 +2,10 @@
 
 set -e
 
-mkdir -p data
+
 touch data/queue.sqlite
 npm i
 
 cat ./sql/setup.sql | sqlite3 data/queue.sqlite
 
-bash ./scripts/build_web.sh
\ No newline at end of file
+bash ./scripts/build_web.sh
diff --git a/scripts/train.sh b/scripts/train.sh
index 0ee2565..e98840d 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -28,8 +28,8 @@ NAME=$(echo $JSON | jq -r '.name')
 FILEPATH=$(echo $JSON | jq -r '.path')
 
 DOWNLOAD="${YOLO_WEB_URL}${FILEPATH}"
-DEST="${YOLOv5}${DATASET}.zip"
-UNZIPPED="${YOLOv5}${DATASET}"
+DEST="${DATASETS}/${DATASET}.zip"
+UNZIPPED="${DATASETS}/${DATASET}"
 
 echo "Downloading ${DOWNLOAD}"