diff --git a/exercise.ipynb b/exercise.ipynb
index 879ab3e..ffb1b32 100644
--- a/exercise.ipynb
+++ b/exercise.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "4b4181d6",
+   "id": "3fd3f3cc",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1169a031",
+   "id": "ff40e072",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -41,7 +41,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "214ed3cb",
+   "id": "80b3f5f0",
    "metadata": {},
    "source": [
     "\n",
@@ -54,7 +54,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "95d01880",
+   "id": "d0232b96",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -68,7 +68,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "05f803f1",
+   "id": "95f75250",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -84,7 +84,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e746b8f0",
+   "id": "83063a1f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -102,7 +102,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8f6d18bf",
+   "id": "5cf88f03",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "cc2209ac",
+   "id": "29213ae2",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -130,7 +130,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "246d5d44",
+   "id": "a85fb713",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -155,7 +155,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "baccff88",
+   "id": "eb09c929",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -166,7 +166,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2525561d",
+   "id": "87862a85",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -192,7 +192,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9922897d",
+   "id": "d16e4c25",
    "metadata": {},
    "source": [
     "# Part 2: Using Integrated Gradients to find what the classifier knows\n",
@@ -202,7 +202,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2add0490",
+   "id": "5c0adb99",
    "metadata": {},
    "source": [
     "## Attributions through integrated gradients\n",
@@ -215,7 +215,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4a50f07b",
+   "id": "1cb699da",
    "metadata": {
     "tags": []
    },
@@ -231,7 +231,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "db4b8deb",
+   "id": "b79bd6c0",
    "metadata": {
     "tags": []
    },
@@ -247,7 +247,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "cbc65a0f",
+   "id": "7ad1a754",
    "metadata": {
     "tags": [
      "task"
@@ -268,7 +268,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8ef8eeec",
+   "id": "8408d3ee",
    "metadata": {
     "tags": []
    },
@@ -281,7 +281,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9506b18b",
+   "id": "325ff7e2",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": []
@@ -293,7 +293,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "32097d64",
+   "id": "68d9e107",
    "metadata": {
     "tags": []
    },
@@ -321,7 +321,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0d3b3cc5",
+   "id": "c96f44f5",
    "metadata": {
     "tags": []
    },
@@ -333,7 +333,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b5de1dee",
+   "id": "42fec87e",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -347,7 +347,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c4ea44f9",
+   "id": "14223225",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -360,7 +360,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "dd2ab31e",
+   "id": "f666dc63",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -384,7 +384,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f996d266",
+   "id": "06dcde16",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -398,7 +398,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5c4eb730",
+   "id": "51832dda",
    "metadata": {},
    "source": [
     "\n",
@@ -424,7 +424,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a4a819f1",
+   "id": "c553eda1",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h4>Task 2.3: Use random noise as a baseline</h4>\n",
@@ -436,7 +436,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2de84b61",
+   "id": "1fdafaac",
    "metadata": {
     "tags": [
      "task"
@@ -456,7 +456,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2f5102f5",
+   "id": "3fc419b6",
    "metadata": {
     "tags": []
    },
@@ -470,7 +470,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2ab73887",
+   "id": "4e9b9f26",
    "metadata": {
     "tags": [
      "task"
@@ -492,7 +492,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9ba8db53",
+   "id": "daeaa65c",
    "metadata": {
     "tags": []
    },
@@ -508,7 +508,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ad636e18",
+   "id": "6cfbd572",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h2>BONUS Task: Using different attributions.</h2>\n",
@@ -522,7 +522,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a2f58ea0",
+   "id": "2dc92a5c",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 2</h2>\n",
@@ -542,7 +542,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2d8ba19e",
+   "id": "c0727f2f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -570,7 +570,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4710cc4e",
+   "id": "147a10f1",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -593,7 +593,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e53413fc",
+   "id": "0b789be2",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -625,7 +625,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "de51e24c",
+   "id": "460878cc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -640,7 +640,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "640b462d",
+   "id": "40b1944f",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -661,7 +661,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d8a63d51",
+   "id": "dc70737d",
    "metadata": {
     "tags": []
    },
@@ -676,7 +676,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "842a622b",
+   "id": "6bd563e2",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -693,7 +693,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "46352b5e",
+   "id": "02d110f4",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -707,7 +707,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a254958b",
+   "id": "955d9981",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -718,7 +718,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e708acc0",
+   "id": "7f71dbdc",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -728,7 +728,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "507389af",
+   "id": "bd0d99c9",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -746,7 +746,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f0895191",
+   "id": "cb6bf33f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -758,7 +758,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "93be09da",
+   "id": "803dad9e",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -777,7 +777,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7be2896e",
+   "id": "1ed827ff",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -786,7 +786,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7082aedf",
+   "id": "5166c91e",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -802,7 +802,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "121b201a",
+   "id": "756f9a51",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -811,7 +811,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d60ff4f5",
+   "id": "625bb412",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -823,7 +823,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1c41c3d0",
+   "id": "8efeda25",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -838,7 +838,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7dc2bc5b",
+   "id": "613e2c1f",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -852,7 +852,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "83e4433e",
+   "id": "971e4622",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -866,7 +866,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "887dcc4e",
+   "id": "d86d0ea1",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -886,7 +886,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "86412c93",
+   "id": "ee7e26ce",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -910,7 +910,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "67aa5e42",
+   "id": "6b0e8161",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -920,7 +920,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7a9574f1",
+   "id": "854f274b",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -942,7 +942,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e694f3fa",
+   "id": "7783da15",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -954,7 +954,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3166f59c",
+   "id": "feb2fe5d",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -1065,7 +1065,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5688f728",
+   "id": "5809a842",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1077,7 +1077,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d9ba276a",
+   "id": "59ea06d6",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1090,7 +1090,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7a617979",
+   "id": "86c8ae57",
    "metadata": {
     "tags": []
    },
@@ -1105,7 +1105,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "83a57e24",
+   "id": "8316db9c",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1117,7 +1117,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7cfef8b9",
+   "id": "d9fbc729",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1136,7 +1136,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d8539b97",
+   "id": "e2a81fb3",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1145,7 +1145,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9b0e218e",
+   "id": "e039a039",
    "metadata": {
     "tags": []
    },
@@ -1161,7 +1161,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e4172419",
+   "id": "7f4210fd",
    "metadata": {
     "tags": []
    },
@@ -1171,7 +1171,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "dde54fef",
+   "id": "faf3eac1",
    "metadata": {
     "tags": []
    },
@@ -1188,7 +1188,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "bd522c56",
+   "id": "b56b0ac0",
    "metadata": {
     "title": "Loading the test dataset"
    },
@@ -1208,7 +1208,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8add0043",
+   "id": "e0ded76f",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1220,7 +1220,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9dbf8de3",
+   "id": "fdcd0b4c",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1233,7 +1233,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "92d62851",
+   "id": "a0a01596",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1243,7 +1243,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f53dea12",
+   "id": "5088af03",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1261,7 +1261,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e0c33dd8",
+   "id": "efe9174b",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -1297,7 +1297,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "00471bff",
+   "id": "c87c89df",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1309,7 +1309,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "72dc24f2",
+   "id": "7db32745",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1319,7 +1319,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "749aaa82",
+   "id": "ed5aafe5",
    "metadata": {
     "tags": []
    },
@@ -1334,7 +1334,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b2f7bd72",
+   "id": "cdba36a8",
    "metadata": {
     "tags": []
    },
@@ -1345,7 +1345,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "873b0840",
+   "id": "de504515",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1359,7 +1359,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4553c214",
+   "id": "d460f4eb",
    "metadata": {
     "tags": []
    },
@@ -1374,7 +1374,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "12113757",
+   "id": "59041d52",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1389,7 +1389,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2afb5f61",
+   "id": "752c4ee3",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -1411,7 +1411,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b63644b4",
+   "id": "1a401326",
    "metadata": {
     "lines_to_next_cell": 1,
     "title": "Another visualization function"
@@ -1441,7 +1441,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "fe366563",
+   "id": "affc1177",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1457,7 +1457,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "dcae89a4",
+   "id": "194ac43d",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1473,7 +1473,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f2a0dae2",
+   "id": "f54356bc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1488,7 +1488,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "67ef87b3",
+   "id": "473e32d8",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1512,7 +1512,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "832a5a11",
+   "id": "0d29cfae",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1525,7 +1525,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3296408d",
+   "id": "3d2d9a2d",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1558,7 +1558,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ef3eedbb",
+   "id": "29cd4445",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1574,7 +1574,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "81df49f2",
+   "id": "d9a0f9e5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1597,7 +1597,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9719086f",
+   "id": "f508f4cc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1611,7 +1611,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c156966d",
+   "id": "31527df5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1628,7 +1628,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "37fc0f75",
+   "id": "2f77a1be",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1637,7 +1637,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "39a664b0",
+   "id": "06b8ef1a",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-warning\"><h3>Questions</h3>\n",
@@ -1649,7 +1649,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9ea3aa28",
+   "id": "a3953322",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 5</h2>\n",
diff --git a/solution.ipynb b/solution.ipynb
index f99e1fe..bd07394 100644
--- a/solution.ipynb
+++ b/solution.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "4b4181d6",
+   "id": "3fd3f3cc",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1169a031",
+   "id": "ff40e072",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -41,7 +41,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "214ed3cb",
+   "id": "80b3f5f0",
    "metadata": {},
    "source": [
     "\n",
@@ -54,7 +54,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "95d01880",
+   "id": "d0232b96",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -68,7 +68,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "05f803f1",
+   "id": "95f75250",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -84,7 +84,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e746b8f0",
+   "id": "83063a1f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -102,7 +102,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8f6d18bf",
+   "id": "5cf88f03",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "cc2209ac",
+   "id": "29213ae2",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -130,7 +130,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3a359874",
+   "id": "b1f1bf12",
    "metadata": {
     "tags": [
      "solution"
@@ -154,7 +154,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "baccff88",
+   "id": "eb09c929",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -165,7 +165,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2525561d",
+   "id": "87862a85",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -191,7 +191,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9922897d",
+   "id": "d16e4c25",
    "metadata": {},
    "source": [
     "# Part 2: Using Integrated Gradients to find what the classifier knows\n",
@@ -201,7 +201,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2add0490",
+   "id": "5c0adb99",
    "metadata": {},
    "source": [
     "## Attributions through integrated gradients\n",
@@ -214,7 +214,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4a50f07b",
+   "id": "1cb699da",
    "metadata": {
     "tags": []
    },
@@ -230,7 +230,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "db4b8deb",
+   "id": "b79bd6c0",
    "metadata": {
     "tags": []
    },
@@ -246,7 +246,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a06d514b",
+   "id": "af7be190",
    "metadata": {
     "tags": [
      "solution"
@@ -270,7 +270,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8ef8eeec",
+   "id": "8408d3ee",
    "metadata": {
     "tags": []
    },
@@ -283,7 +283,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9506b18b",
+   "id": "325ff7e2",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": []
@@ -295,7 +295,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "32097d64",
+   "id": "68d9e107",
    "metadata": {
     "tags": []
    },
@@ -323,7 +323,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0d3b3cc5",
+   "id": "c96f44f5",
    "metadata": {
     "tags": []
    },
@@ -335,7 +335,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b5de1dee",
+   "id": "42fec87e",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -349,7 +349,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c4ea44f9",
+   "id": "14223225",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -362,7 +362,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "dd2ab31e",
+   "id": "f666dc63",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -386,7 +386,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f996d266",
+   "id": "06dcde16",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -400,7 +400,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5c4eb730",
+   "id": "51832dda",
    "metadata": {},
    "source": [
     "\n",
@@ -426,7 +426,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a4a819f1",
+   "id": "c553eda1",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h4>Task 2.3: Use random noise as a baseline</h4>\n",
@@ -438,7 +438,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "73549bdd",
+   "id": "788a207c",
    "metadata": {
     "tags": [
      "solution"
@@ -463,7 +463,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2f5102f5",
+   "id": "3fc419b6",
    "metadata": {
     "tags": []
    },
@@ -477,7 +477,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6569d29f",
+   "id": "5e86267c",
    "metadata": {
     "tags": [
      "solution"
@@ -504,7 +504,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9ba8db53",
+   "id": "daeaa65c",
    "metadata": {
     "tags": []
    },
@@ -520,7 +520,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ad636e18",
+   "id": "6cfbd572",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h2>BONUS Task: Using different attributions.</h2>\n",
@@ -534,7 +534,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a2f58ea0",
+   "id": "2dc92a5c",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 2</h2>\n",
@@ -554,7 +554,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2d8ba19e",
+   "id": "c0727f2f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -582,7 +582,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4710cc4e",
+   "id": "147a10f1",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -605,7 +605,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e53413fc",
+   "id": "0b789be2",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -637,7 +637,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "de51e24c",
+   "id": "460878cc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -652,7 +652,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b443096c",
+   "id": "5cf75884",
    "metadata": {
     "tags": [
      "solution"
@@ -669,7 +669,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d8a63d51",
+   "id": "dc70737d",
    "metadata": {
     "tags": []
    },
@@ -684,7 +684,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "842a622b",
+   "id": "6bd563e2",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -701,7 +701,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "009ba6a8",
+   "id": "6fa80433",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -715,7 +715,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a254958b",
+   "id": "955d9981",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -726,7 +726,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e708acc0",
+   "id": "7f71dbdc",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -736,7 +736,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "507389af",
+   "id": "bd0d99c9",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -754,7 +754,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f0895191",
+   "id": "cb6bf33f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -766,7 +766,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "93be09da",
+   "id": "803dad9e",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -785,7 +785,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7be2896e",
+   "id": "1ed827ff",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -794,7 +794,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7082aedf",
+   "id": "5166c91e",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -810,7 +810,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "121b201a",
+   "id": "756f9a51",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -819,7 +819,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d60ff4f5",
+   "id": "625bb412",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -831,7 +831,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1c41c3d0",
+   "id": "8efeda25",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -846,7 +846,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7dc2bc5b",
+   "id": "613e2c1f",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -860,7 +860,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "83e4433e",
+   "id": "971e4622",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -874,7 +874,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "887dcc4e",
+   "id": "d86d0ea1",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -894,7 +894,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "86412c93",
+   "id": "ee7e26ce",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -918,7 +918,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "67aa5e42",
+   "id": "6b0e8161",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -928,7 +928,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7a9574f1",
+   "id": "854f274b",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -950,7 +950,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e694f3fa",
+   "id": "7783da15",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -962,7 +962,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f7d8ecce",
+   "id": "14812a7c",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": [
@@ -1032,7 +1032,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5688f728",
+   "id": "5809a842",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1044,7 +1044,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d9ba276a",
+   "id": "59ea06d6",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1057,7 +1057,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7a617979",
+   "id": "86c8ae57",
    "metadata": {
     "tags": []
    },
@@ -1072,7 +1072,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "83a57e24",
+   "id": "8316db9c",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1084,7 +1084,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7cfef8b9",
+   "id": "d9fbc729",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1103,7 +1103,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d8539b97",
+   "id": "e2a81fb3",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1112,7 +1112,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9b0e218e",
+   "id": "e039a039",
    "metadata": {
     "tags": []
    },
@@ -1128,7 +1128,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e4172419",
+   "id": "7f4210fd",
    "metadata": {
     "tags": []
    },
@@ -1138,7 +1138,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "dde54fef",
+   "id": "faf3eac1",
    "metadata": {
     "tags": []
    },
@@ -1155,7 +1155,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "bd522c56",
+   "id": "b56b0ac0",
    "metadata": {
     "title": "Loading the test dataset"
    },
@@ -1175,7 +1175,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8add0043",
+   "id": "e0ded76f",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1187,7 +1187,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9dbf8de3",
+   "id": "fdcd0b4c",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1200,7 +1200,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "92d62851",
+   "id": "a0a01596",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1210,7 +1210,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f53dea12",
+   "id": "5088af03",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1228,7 +1228,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1226f193",
+   "id": "b4cb730b",
    "metadata": {
     "tags": [
      "solution"
@@ -1265,7 +1265,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "00471bff",
+   "id": "c87c89df",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1277,7 +1277,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "72dc24f2",
+   "id": "7db32745",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1287,7 +1287,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "749aaa82",
+   "id": "ed5aafe5",
    "metadata": {
     "tags": []
    },
@@ -1302,7 +1302,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b2f7bd72",
+   "id": "cdba36a8",
    "metadata": {
     "tags": []
    },
@@ -1313,7 +1313,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "873b0840",
+   "id": "de504515",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1327,7 +1327,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4553c214",
+   "id": "d460f4eb",
    "metadata": {
     "tags": []
    },
@@ -1342,7 +1342,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "12113757",
+   "id": "59041d52",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1357,7 +1357,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2afb5f61",
+   "id": "752c4ee3",
    "metadata": {
     "lines_to_next_cell": 1
    },
@@ -1379,7 +1379,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b63644b4",
+   "id": "1a401326",
    "metadata": {
     "lines_to_next_cell": 1,
     "title": "Another visualization function"
@@ -1409,7 +1409,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "fe366563",
+   "id": "affc1177",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1425,7 +1425,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "dcae89a4",
+   "id": "194ac43d",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1441,7 +1441,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f2a0dae2",
+   "id": "f54356bc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1456,7 +1456,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "67ef87b3",
+   "id": "473e32d8",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1480,7 +1480,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "832a5a11",
+   "id": "0d29cfae",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1493,7 +1493,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3296408d",
+   "id": "3d2d9a2d",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1526,7 +1526,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ef3eedbb",
+   "id": "29cd4445",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1542,7 +1542,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "81df49f2",
+   "id": "d9a0f9e5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1565,7 +1565,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9719086f",
+   "id": "f508f4cc",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1579,7 +1579,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c156966d",
+   "id": "31527df5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1596,7 +1596,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0089c05a",
+   "id": "94b216f2",
    "metadata": {
     "tags": [
      "solution"
@@ -1619,7 +1619,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "37fc0f75",
+   "id": "2f77a1be",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1628,7 +1628,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "39a664b0",
+   "id": "06b8ef1a",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-warning\"><h3>Questions</h3>\n",
@@ -1640,7 +1640,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9ea3aa28",
+   "id": "a3953322",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 5</h2>\n",
@@ -1658,7 +1658,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "bbfc39f2",
+   "id": "4c2eb6f3",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -1673,7 +1673,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3dea95db",
+   "id": "b972033b",
    "metadata": {
     "tags": [
      "solution"