diff --git a/Exercise_BrainMRI_AgeClassif_DL_Work.mlx b/Exercise_BrainMRI_AgeClassif_DL_Work.mlx
index 78c6150..a6be956 100644
Binary files a/Exercise_BrainMRI_AgeClassif_DL_Work.mlx and b/Exercise_BrainMRI_AgeClassif_DL_Work.mlx differ
diff --git a/prepare2DImageDataset.m b/prepare2DImageDataset.m
index d55fa49..29f132b 100644
--- a/prepare2DImageDataset.m
+++ b/prepare2DImageDataset.m
@@ -17,14 +17,7 @@
 end
 
 % Create classification image set folders
-if ~isempty(dstPath)
-    i = 1;
-    t = dstPath;
-    while exist(t,'dir')
-        t = dstPath+"_"+i;
-        i = i+1;
-    end
-    dstPath = i;
+if ~isempty(dstPath)   
     assert(~exist(dstPath,'dir'));
     mkdir(dstPath);
 end
@@ -153,4 +146,4 @@
 end
 end
 
-% Copyright 2020-2021 The MathWorks, Inc.
\ No newline at end of file
+% Copyright 2020-2021 The MathWorks, Inc.