diff --git a/Lachlan_SMOTE_GAN.ipynb b/Lachlan_SMOTE_GAN.ipynb
index 73d5824..d576710 100644
--- a/Lachlan_SMOTE_GAN.ipynb
+++ b/Lachlan_SMOTE_GAN.ipynb
@@ -5,6 +5,14 @@
    "id": "d7973263",
    "metadata": {},
    "source": [
+    "# Data augmentation using SMOTE-GAN\n",
+    "\n",
+    "### Ehsan Farahbakhsh, Nathan Wake, R. Ditmar Müller\n",
+    "\n",
+    "*EarthByte Group, School of Geosciences, University of Sydney, NSW 2006, Australia*\n",
+    "\n",
+    "This notebook is crafted to augment positive data, specifically known mineral occurrences, within the framework of mineral prospectivity mapping. It utilizes a hybrid approach that integrates the Synthetic Minority Oversampling Technique (SMOTE) with Generative Adversarial Networks (GAN) to enhance the dataset.\n",
+    "\n",
     "### Libraries"
    ]
   },
@@ -247,7 +255,7 @@
     "    Xy_train_cat_ = pd.concat([Xy_train_cat, Xy_train_cat_positive_mode_]).reset_index(drop=True)\n",
     "    smote_gan_samples = pd.concat([smote_gan_samples, Xy_train_cat_], axis=1).reset_index(drop=True)\n",
     "    smote_gan_samples = smote_gan_samples[features_label_lst]\n",
-    "    smote_gan_samples.to_csv(f'./Datasets/Outputs/smote_gan_important_01_{i+1}.csv', index=False)"
+    "    smote_gan_samples.to_csv(f'./Datasets/Outputs/smote_gan_important_{i+1}.csv', index=False)"
    ]
   },
   {