diff --git a/DLWMLS/__main__.py b/DLWMLS/__main__.py
index 30d50d2..8c5bf07 100644
--- a/DLWMLS/__main__.py
+++ b/DLWMLS/__main__.py
@@ -120,7 +120,7 @@ def main() -> None:
         "-d",
         type=str,
         required=False,
-        default="002",
+        default="003",
         help="[nnUnet Arg] Dataset with which you would like to predict. You can specify either dataset name or id",
     )
     parser.add_argument(
diff --git a/DLWMLS/__pycache__/__init__.cpython-312.pyc b/DLWMLS/__pycache__/__init__.cpython-312.pyc
index b9a5f83..1f035aa 100644
Binary files a/DLWMLS/__pycache__/__init__.cpython-312.pyc and b/DLWMLS/__pycache__/__init__.cpython-312.pyc differ
diff --git a/DLWMLS/__pycache__/__main__.cpython-312.pyc b/DLWMLS/__pycache__/__main__.cpython-312.pyc
index ea558c2..3890dec 100644
Binary files a/DLWMLS/__pycache__/__main__.cpython-312.pyc and b/DLWMLS/__pycache__/__main__.cpython-312.pyc differ
diff --git a/DLWMLS/__pycache__/utils.cpython-312.pyc b/DLWMLS/__pycache__/utils.cpython-312.pyc
index c91cb06..fc10d70 100644
Binary files a/DLWMLS/__pycache__/utils.cpython-312.pyc and b/DLWMLS/__pycache__/utils.cpython-312.pyc differ
diff --git a/DLWMLS/utils.py b/DLWMLS/utils.py
index 719d295..cb3bad8 100644
--- a/DLWMLS/utils.py
+++ b/DLWMLS/utils.py
@@ -36,6 +36,7 @@ def rename_and_copy_files(src_folder: str, des_folder: str) -> Tuple[dict, dict]
         new_name = os.path.join(des_folder, rename_file)
         shutil.copy2(old_name, new_name)
         rename_dict[filename] = rename_file
-        rename_back_dict[rename_back] = "DLWMLS_mask_" + filename
+        rename_back_dict[rename_back] = filename.split(".nii")[0] + "_WMLS.nii.gz"
+        print("%s will be renamed to %s" % (old_name, filename.split(".nii")[0] + "_WMLS.nii.gz"))
 
     return rename_dict, rename_back_dict