diff --git a/.gitignore b/.gitignore
index ff1fd384..7cca0d67 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,8 +7,11 @@ __pycache__/
 
 # Outputs
 output/
-data/classification_dataset
-data/output_images/
+outputs/
+dataset/
+!docs/dataset/
+assets/classification_dataset
+assets/output_images/
 database/data/
 
 # Virtual environments
diff --git a/docs/dataset.md b/docs/dataset/CropAndWeed.md
similarity index 97%
rename from docs/dataset.md
rename to docs/dataset/CropAndWeed.md
index 61cd1061..293774f3 100644
--- a/docs/dataset.md
+++ b/docs/dataset/CropAndWeed.md
@@ -1,5 +1,5 @@
+# CropAndWeed Dataset
 
-# CropAndWeed Dataset Documentation
 ## Dataset Overview
 - **Source Link**: [https://github.com/cropandweed/cropandweed-dataset](https://github.com/cropandweed/cropandweed-dataset)
 - **Paper**: [The CropAndWeed Dataset: A Multi-Modal Learning Approach for Efficient Crop and Weed Manipulation](https://openaccess.thecvf.com/content/WACV2023/papers/Steininger_The_CropAndWeed_Dataset_A_Multi-Modal_Learning_Approach_for_Efficient_Crop_WACV_2023_paper.pdf) (WACV 2023)
diff --git a/docs/dataset_preparation.md b/docs/dataset/dataset_preparation.md
similarity index 99%
rename from docs/dataset_preparation.md
rename to docs/dataset/dataset_preparation.md
index d2c66f55..f0ba00a8 100644
--- a/docs/dataset_preparation.md
+++ b/docs/dataset/dataset_preparation.md
@@ -1,4 +1,7 @@
-# Dataset Preparation
+# CropAndWeed Dataset Preparation
+
+## Overview
+...
 
 > **Note:** Make sure to install and enable [Git LFS](https://git-lfs.com/) before extracting the dataset.
 > Run:
@@ -65,7 +68,7 @@ python dataset/preparation/decompress_dataset.py
 
 ---
 
-## Overview
+## Dataset Description
 
 The **CropAndWeed** dataset contains agricultural images with multi-level annotations for **crop and weed classification**.