diff --git a/MNIST_data/MNIST/processed/test.pt b/MNIST_data/MNIST/processed/test.pt
new file mode 100644
index 00000000..b152901e
Binary files /dev/null and b/MNIST_data/MNIST/processed/test.pt differ
diff --git a/MNIST_data/MNIST/processed/training.pt b/MNIST_data/MNIST/processed/training.pt
new file mode 100644
index 00000000..cdaa723b
Binary files /dev/null and b/MNIST_data/MNIST/processed/training.pt differ
diff --git a/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte b/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte
new file mode 100644
index 00000000..1170b2ca
Binary files /dev/null and b/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte differ
diff --git a/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte.gz b/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte.gz
new file mode 100644
index 00000000..5ace8ea9
Binary files /dev/null and b/MNIST_data/MNIST/raw/t10k-images-idx3-ubyte.gz differ
diff --git a/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte b/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte
new file mode 100644
index 00000000..d1c3a970
Binary files /dev/null and b/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte differ
diff --git a/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte.gz b/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte.gz
new file mode 100644
index 00000000..a7e14154
Binary files /dev/null and b/MNIST_data/MNIST/raw/t10k-labels-idx1-ubyte.gz differ
diff --git a/MNIST_data/MNIST/raw/train-images-idx3-ubyte b/MNIST_data/MNIST/raw/train-images-idx3-ubyte
new file mode 100644
index 00000000..bbce2765
Binary files /dev/null and b/MNIST_data/MNIST/raw/train-images-idx3-ubyte differ
diff --git a/MNIST_data/MNIST/raw/train-images-idx3-ubyte.gz b/MNIST_data/MNIST/raw/train-images-idx3-ubyte.gz
new file mode 100644
index 00000000..b50e4b6b
Binary files /dev/null and b/MNIST_data/MNIST/raw/train-images-idx3-ubyte.gz differ
diff --git a/MNIST_data/MNIST/raw/train-labels-idx1-ubyte b/MNIST_data/MNIST/raw/train-labels-idx1-ubyte
new file mode 100644
index 00000000..d6b4c5db
Binary files /dev/null and b/MNIST_data/MNIST/raw/train-labels-idx1-ubyte differ
diff --git a/MNIST_data/MNIST/raw/train-labels-idx1-ubyte.gz b/MNIST_data/MNIST/raw/train-labels-idx1-ubyte.gz
new file mode 100644
index 00000000..707a576b
Binary files /dev/null and b/MNIST_data/MNIST/raw/train-labels-idx1-ubyte.gz differ
diff --git a/Part 2 - Neural Networks in PyTorch.ipynb b/Part 2 - Neural Networks in PyTorch.ipynb
index 6d139c0f..55138d1f 100644
--- a/Part 2 - Neural Networks in PyTorch.ipynb	
+++ b/Part 2 - Neural Networks in PyTorch.ipynb	
@@ -44,7 +44,7 @@
    "source": [
     "# Define a transform to normalize the data\n",
     "transform = transforms.Compose([transforms.ToTensor(),\n",
-    "                              transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),\n",
+    "                              transforms.Normalize([0.5],[0.5]),\n",
     "                             ])\n",
     "# Download and load the training data\n",
     "trainset = datasets.MNIST('MNIST_data/', download=True, train=True, transform=transform)\n",
@@ -415,4 +415,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file
diff --git a/Part 4 - Fashion-MNIST Exercise.ipynb b/Part 4 - Fashion-MNIST Exercise.ipynb
index 2928d9b3..eabeac1b 100644
--- a/Part 4 - Fashion-MNIST Exercise.ipynb	
+++ b/Part 4 - Fashion-MNIST Exercise.ipynb	
@@ -27,7 +27,7 @@
     "\n",
     "# Define a transform to normalize the data\n",
     "transform = transforms.Compose([transforms.ToTensor(),\n",
-    "                                transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])\n",
+    "                                transforms.Normalize([0.5], [0.5]])\n",
     "# Download and load the training data\n",
     "trainset = datasets.FashionMNIST('F_MNIST_data/', download=True, train=True, transform=transform)\n",
     "trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)\n",
@@ -187,4 +187,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file
diff --git a/__pycache__/helper.cpython-37.pyc b/__pycache__/helper.cpython-37.pyc
new file mode 100644
index 00000000..2a4c559e
Binary files /dev/null and b/__pycache__/helper.cpython-37.pyc differ