diff --git a/baselines/.ipynb_checkpoints/MNIST_baseline-checkpoint.ipynb b/baselines/.ipynb_checkpoints/MNIST_baseline-checkpoint.ipynb
index 81b468a2a1636919938500f286957ae7cd904275..781607ee7952fb7027789701424278467d3289ac 100644
--- a/baselines/.ipynb_checkpoints/MNIST_baseline-checkpoint.ipynb
+++ b/baselines/.ipynb_checkpoints/MNIST_baseline-checkpoint.ipynb
@@ -59,30 +59,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "--2020-05-17 00:00:14--  https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/test.zip\n",
-      "Resolving s3.eu-central-1.wasabisys.com (s3.eu-central-1.wasabisys.com)... 130.117.252.16, 130.117.252.12, 130.117.252.13, ...\n",
-      "Connecting to s3.eu-central-1.wasabisys.com (s3.eu-central-1.wasabisys.com)|130.117.252.16|:443... connected.\n",
-      "HTTP request sent, awaiting response... 200 OK\n",
-      "Length: 2202644 (2.1M) [application/zip]\n",
-      "Saving to: ‘test.zip’\n",
-      "\n",
-      "test.zip            100%[===================>]   2.10M  64.3KB/s    in 29s     \n",
-      "\n",
-      "2020-05-17 00:00:44 (74.3 KB/s) - ‘test.zip’ saved [2202644/2202644]\n",
-      "\n",
-      "Archive:  test.zip\n",
-      "  inflating: test.csv                \n",
-      "rm: cannot remove 'train.zip': No such file or directory\n"
-     ]
-    }
-   ],
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "#Donwload the datasets\n",
     "!rm -rf data\n",
@@ -90,6 +69,7 @@
     "!wget https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/test.zip\n",
     "!wget https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/train.zip\n",
     "!unzip test.zip\n",
+    "!unzip test.zip\n",
     "!mv test.csv data/test.csv\n",
     "!mv train.csv data/train.csv\n",
     "!rm test.zip train.zip"
diff --git a/baselines/MNIST_baseline.ipynb b/baselines/MNIST_baseline.ipynb
index 81b468a2a1636919938500f286957ae7cd904275..781607ee7952fb7027789701424278467d3289ac 100644
--- a/baselines/MNIST_baseline.ipynb
+++ b/baselines/MNIST_baseline.ipynb
@@ -59,30 +59,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "--2020-05-17 00:00:14--  https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/test.zip\n",
-      "Resolving s3.eu-central-1.wasabisys.com (s3.eu-central-1.wasabisys.com)... 130.117.252.16, 130.117.252.12, 130.117.252.13, ...\n",
-      "Connecting to s3.eu-central-1.wasabisys.com (s3.eu-central-1.wasabisys.com)|130.117.252.16|:443... connected.\n",
-      "HTTP request sent, awaiting response... 200 OK\n",
-      "Length: 2202644 (2.1M) [application/zip]\n",
-      "Saving to: ‘test.zip’\n",
-      "\n",
-      "test.zip            100%[===================>]   2.10M  64.3KB/s    in 29s     \n",
-      "\n",
-      "2020-05-17 00:00:44 (74.3 KB/s) - ‘test.zip’ saved [2202644/2202644]\n",
-      "\n",
-      "Archive:  test.zip\n",
-      "  inflating: test.csv                \n",
-      "rm: cannot remove 'train.zip': No such file or directory\n"
-     ]
-    }
-   ],
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "#Donwload the datasets\n",
     "!rm -rf data\n",
@@ -90,6 +69,7 @@
     "!wget https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/test.zip\n",
     "!wget https://s3.eu-central-1.wasabisys.com/aicrowd-practice-challenges/public/mnist/v0.1/train.zip\n",
     "!unzip test.zip\n",
+    "!unzip test.zip\n",
     "!mv test.csv data/test.csv\n",
     "!mv train.csv data/train.csv\n",
     "!rm test.zip train.zip"