diff --git a/src/dataset_preparer.py b/src/dataset_preparer.py
index a77b198..1953122 100644
--- a/src/dataset_preparer.py
+++ b/src/dataset_preparer.py
@@ -22,6 +22,7 @@ def __init__(self, dataset_name, files):
         self.dataset_name = dataset_name
         self.files = files
         self.dataset_dir = f"data/{self.dataset_name}/"
+        self.filename = self.files[0].split('/')[-1].split('.')[0]
         logging.basicConfig(level=logging.INFO)
 
     def download(self):