diff --git a/exercise.ipynb b/exercise.ipynb
index ab54c2d..50d5adb 100644
--- a/exercise.ipynb
+++ b/exercise.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "0e12b745",
+   "id": "be4e7a97",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a5a4c73b",
+   "id": "a1e6c3cd",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -41,7 +41,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "caee6e2d",
+   "id": "bc9d8f39",
    "metadata": {},
    "source": [
     "\n",
@@ -54,7 +54,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "60be885c",
+   "id": "5a28affe",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -68,7 +68,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "17dfe28f",
+   "id": "3fb92e01",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -84,7 +84,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e22f0615",
+   "id": "225052d7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -102,7 +102,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b69e6a45",
+   "id": "a878ce13",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "37df023f",
+   "id": "1f8d5401",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -130,7 +130,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a98e6cb7",
+   "id": "19f50056",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -155,7 +155,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "250e04ae",
+   "id": "6dd3913c",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -166,7 +166,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c0bf76da",
+   "id": "060a05f5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -195,7 +195,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d3657082",
+   "id": "2616d093",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -212,7 +212,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d7d6d20b",
+   "id": "76fae027",
    "metadata": {},
    "source": [
     "# Part 2: Using Integrated Gradients to find what the classifier knows\n",
@@ -222,7 +222,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2651461e",
+   "id": "5284eaf7",
    "metadata": {},
    "source": [
     "## Attributions through integrated gradients\n",
@@ -235,7 +235,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0a6c14a4",
+   "id": "57eb7e30",
    "metadata": {
     "tags": []
    },
@@ -253,7 +253,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ea5ae280",
+   "id": "5dd749a7",
    "metadata": {
     "tags": []
    },
@@ -269,7 +269,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c3089484",
+   "id": "7d56e7b9",
    "metadata": {
     "tags": [
      "task"
@@ -290,7 +290,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "48053a52",
+   "id": "f1942f36",
    "metadata": {
     "tags": []
    },
@@ -303,7 +303,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a171cfd0",
+   "id": "7e5d1815",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": []
@@ -315,7 +315,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "5f1fa7f4",
+   "id": "d5103ffb",
    "metadata": {
     "tags": []
    },
@@ -343,7 +343,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6d30759d",
+   "id": "858a7c14",
    "metadata": {
     "tags": []
    },
@@ -356,7 +356,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "201452d4",
+   "id": "78c186ef",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -370,7 +370,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "212b9a38",
+   "id": "5e3aa105",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -383,7 +383,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ac0d72aa",
+   "id": "8171ef5a",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -408,7 +408,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "445763f0",
+   "id": "0b65a750",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -422,7 +422,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d786f7a9",
+   "id": "06d9fccd",
    "metadata": {},
    "source": [
     "\n",
@@ -448,7 +448,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ee1ed4c2",
+   "id": "1cf3195f",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h4>Task 2.3: Use random noise as a baseline</h4>\n",
@@ -460,7 +460,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a2d9d5ff",
+   "id": "fbefdb10",
    "metadata": {
     "tags": [
      "task"
@@ -482,7 +482,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1ba578d2",
+   "id": "e31bd3cf",
    "metadata": {
     "tags": []
    },
@@ -496,7 +496,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "da1b4a9a",
+   "id": "98de5d82",
    "metadata": {
     "tags": [
      "task"
@@ -520,7 +520,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2f015ba4",
+   "id": "290bdbf3",
    "metadata": {
     "tags": []
    },
@@ -536,7 +536,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e087f05d",
+   "id": "659c5758",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h2>BONUS Task: Using different attributions.</h2>\n",
@@ -550,7 +550,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ae76eb75",
+   "id": "b93071a2",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 2</h2>\n",
@@ -570,7 +570,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b2ec6177",
+   "id": "e9f2b7ae",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -598,7 +598,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "11737f5e",
+   "id": "92223fdc",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -621,7 +621,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "382b5caf",
+   "id": "5a72db0b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -653,7 +653,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "09c1a43a",
+   "id": "01335c40",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -668,7 +668,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "966f9091",
+   "id": "8d0f0203",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -689,7 +689,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e254f618",
+   "id": "186f40a5",
    "metadata": {
     "tags": []
    },
@@ -704,7 +704,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9817ee9d",
+   "id": "af4cf127",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -721,7 +721,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "199fadb0",
+   "id": "94f07b70",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -735,7 +735,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5c1b84fa",
+   "id": "c2a072a4",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -746,7 +746,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "35a1b628",
+   "id": "f7bb217b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -756,7 +756,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0741cd51",
+   "id": "dfdd1272",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -774,7 +774,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9ea04106",
+   "id": "7ddd581a",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -786,7 +786,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "df0aa4e4",
+   "id": "2f7fe43c",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -805,7 +805,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "888fdbee",
+   "id": "f01095a7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -814,7 +814,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "3c6e2c10",
+   "id": "bad091aa",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -830,7 +830,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c959fb2f",
+   "id": "e49c55e0",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -839,7 +839,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e7249572",
+   "id": "af990b9a",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -851,7 +851,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "17b8c4b3",
+   "id": "291f2062",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -864,7 +864,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "fedb48d8",
+   "id": "919aea73",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -878,7 +878,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2ddc0100",
+   "id": "26f2a9c1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -890,7 +890,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4d4d2810",
+   "id": "54780ac3",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -910,7 +910,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "432f2f53",
+   "id": "e7187b29",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -934,7 +934,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9d6397cf",
+   "id": "78521f46",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -946,7 +946,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "779bcf13",
+   "id": "036e6086",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -968,7 +968,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "752616b6",
+   "id": "3b756b7a",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -980,7 +980,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c95ad770",
+   "id": "20847a1b",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -1091,7 +1091,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a7cdf035",
+   "id": "78403aaa",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1103,7 +1103,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4bd99022",
+   "id": "9bb6c797",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1119,7 +1119,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2ca0425e",
+   "id": "c32cbd8b",
    "metadata": {
     "tags": []
    },
@@ -1134,7 +1134,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8ed754f6",
+   "id": "4c86dd42",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1146,7 +1146,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c90cd932",
+   "id": "8f728ded",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1168,7 +1168,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "23d27b74",
+   "id": "028d3bbc",
    "metadata": {
     "tags": []
    },
@@ -1184,7 +1184,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0afeb5a4",
+   "id": "b63aac86",
    "metadata": {
     "tags": []
    },
@@ -1194,7 +1194,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "61e418f1",
+   "id": "b0ad5935",
    "metadata": {
     "tags": []
    },
@@ -1211,7 +1211,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7b6c2411",
+   "id": "0cf2b9a5",
    "metadata": {
     "title": "Loading the test dataset"
    },
@@ -1231,7 +1231,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "26cd8983",
+   "id": "35cc9b35",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1243,7 +1243,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f1f5cc1e",
+   "id": "4146969c",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1256,7 +1256,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f3cd6599",
+   "id": "f99f676c",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1266,7 +1266,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "cefcba31",
+   "id": "a7450339",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1284,7 +1284,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8b706e0a",
+   "id": "58973345",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -1320,7 +1320,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "bfda4b7f",
+   "id": "20af6915",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1332,7 +1332,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a5b2f903",
+   "id": "4543762f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1345,7 +1345,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "935009d9",
+   "id": "d0b8a2ec",
    "metadata": {
     "tags": []
    },
@@ -1360,7 +1360,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7683e70e",
+   "id": "61e58d7f",
    "metadata": {
     "tags": []
    },
@@ -1371,7 +1371,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3532d6ed",
+   "id": "b8dc4640",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1385,7 +1385,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c8c90e6b",
+   "id": "ed9b7104",
    "metadata": {
     "tags": []
    },
@@ -1400,7 +1400,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c4c3eff4",
+   "id": "0add70ab",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1415,7 +1415,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1eff97c0",
+   "id": "274d8226",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1436,7 +1436,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c1281384",
+   "id": "b6b54bbe",
    "metadata": {
     "title": "Another visualization function"
    },
@@ -1465,7 +1465,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "42cf09ad",
+   "id": "979ca20f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1481,7 +1481,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "30ea4f96",
+   "id": "b779b535",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1497,7 +1497,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6768c3d0",
+   "id": "c44f0cf1",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1512,7 +1512,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0287a687",
+   "id": "f402bb74",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1526,7 +1526,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6459abb5",
+   "id": "12542a6d",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1546,7 +1546,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ca40f5a7",
+   "id": "8d3e6d86",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1582,7 +1582,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "94276a21",
+   "id": "ca1d1242",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1596,7 +1596,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a7597f89",
+   "id": "6f530e82",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1612,7 +1612,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9cb53f3f",
+   "id": "5904b20b",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1628,7 +1628,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6a43a957",
+   "id": "72ca6d87",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1651,7 +1651,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "50e69413",
+   "id": "a0baa037",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h3>Task 5.1: Explore the style space</h3>\n",
@@ -1663,7 +1663,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c71b768e",
+   "id": "bc0062db",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1698,12 +1698,12 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a4676c24",
+   "id": "ba428131",
    "metadata": {
     "lines_to_next_cell": 0
    },
    "source": [
-    "<div class=\"alert alert-block alert-info\"><h3>Task 5.1: Adding color to the style space</h3>\n",
+    "<div class=\"alert alert-block alert-info\"><h3>Task 5.2: Adding color to the style space</h3>\n",
     "We know that color is important. Does interpreting the style space as colors help us understand better?\n",
     "\n",
     "Let's use the style space to color the PCA plot.\n",
@@ -1714,7 +1714,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b5838e7c",
+   "id": "2ee2e061",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1741,7 +1741,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2aff4255",
+   "id": "a26903b8",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1755,12 +1755,12 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a80963e3",
+   "id": "7f72dfb5",
    "metadata": {
     "lines_to_next_cell": 0
    },
    "source": [
-    "<div class=\"alert alert-block alert-info\"><h3>Task 5.2: Using the images to color the style space</h3>\n",
+    "<div class=\"alert alert-block alert-info\"><h3>Task 5.3: Using the images to color the style space</h3>\n",
     "Finally, let's just use the colors from the images themselves!\n",
     "The maximum value in the image (since they are \"black-and-color\") can be used as a color!\n",
     "\n",
@@ -1772,7 +1772,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "65e070c3",
+   "id": "be62a09b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1794,7 +1794,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6f5d93bb",
+   "id": "f16385e4",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-warning\"><h3>Questions</h3>\n",
@@ -1806,7 +1806,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1c2bf24d",
+   "id": "e0f59eee",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 5</h2>\n",
@@ -1824,7 +1824,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c5cb1ade",
+   "id": "0bbc6cd2",
    "metadata": {},
    "source": [
     "# Bonus!\n",
diff --git a/solution.ipynb b/solution.ipynb
index 8e26901..d4808b6 100644
--- a/solution.ipynb
+++ b/solution.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "0e12b745",
+   "id": "be4e7a97",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a5a4c73b",
+   "id": "a1e6c3cd",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -41,7 +41,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "caee6e2d",
+   "id": "bc9d8f39",
    "metadata": {},
    "source": [
     "\n",
@@ -54,7 +54,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "60be885c",
+   "id": "5a28affe",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -68,7 +68,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "17dfe28f",
+   "id": "3fb92e01",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -84,7 +84,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e22f0615",
+   "id": "225052d7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -102,7 +102,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b69e6a45",
+   "id": "a878ce13",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "37df023f",
+   "id": "1f8d5401",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -130,7 +130,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9f9053c6",
+   "id": "dc1f9832",
    "metadata": {
     "tags": [
      "solution"
@@ -154,7 +154,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "250e04ae",
+   "id": "6dd3913c",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -165,7 +165,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c0bf76da",
+   "id": "060a05f5",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -194,7 +194,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d3657082",
+   "id": "2616d093",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -211,7 +211,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d7d6d20b",
+   "id": "76fae027",
    "metadata": {},
    "source": [
     "# Part 2: Using Integrated Gradients to find what the classifier knows\n",
@@ -221,7 +221,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2651461e",
+   "id": "5284eaf7",
    "metadata": {},
    "source": [
     "## Attributions through integrated gradients\n",
@@ -234,7 +234,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0a6c14a4",
+   "id": "57eb7e30",
    "metadata": {
     "tags": []
    },
@@ -252,7 +252,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ea5ae280",
+   "id": "5dd749a7",
    "metadata": {
     "tags": []
    },
@@ -268,7 +268,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7eb845d8",
+   "id": "9e77b469",
    "metadata": {
     "tags": [
      "solution"
@@ -292,7 +292,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "48053a52",
+   "id": "f1942f36",
    "metadata": {
     "tags": []
    },
@@ -305,7 +305,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a171cfd0",
+   "id": "7e5d1815",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": []
@@ -317,7 +317,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "5f1fa7f4",
+   "id": "d5103ffb",
    "metadata": {
     "tags": []
    },
@@ -345,7 +345,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6d30759d",
+   "id": "858a7c14",
    "metadata": {
     "tags": []
    },
@@ -358,7 +358,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "201452d4",
+   "id": "78c186ef",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -372,7 +372,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "212b9a38",
+   "id": "5e3aa105",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -385,7 +385,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ac0d72aa",
+   "id": "8171ef5a",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -410,7 +410,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "445763f0",
+   "id": "0b65a750",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -424,7 +424,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d786f7a9",
+   "id": "06d9fccd",
    "metadata": {},
    "source": [
     "\n",
@@ -450,7 +450,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ee1ed4c2",
+   "id": "1cf3195f",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h4>Task 2.3: Use random noise as a baseline</h4>\n",
@@ -462,7 +462,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a3d5d688",
+   "id": "b09d2144",
    "metadata": {
     "tags": [
      "solution"
@@ -488,7 +488,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1ba578d2",
+   "id": "e31bd3cf",
    "metadata": {
     "tags": []
    },
@@ -502,7 +502,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9e7deaa3",
+   "id": "af1e6318",
    "metadata": {
     "tags": [
      "solution"
@@ -532,7 +532,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2f015ba4",
+   "id": "290bdbf3",
    "metadata": {
     "tags": []
    },
@@ -548,7 +548,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e087f05d",
+   "id": "659c5758",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h2>BONUS Task: Using different attributions.</h2>\n",
@@ -562,7 +562,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "ae76eb75",
+   "id": "b93071a2",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 2</h2>\n",
@@ -582,7 +582,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b2ec6177",
+   "id": "e9f2b7ae",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -610,7 +610,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "11737f5e",
+   "id": "92223fdc",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -633,7 +633,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "382b5caf",
+   "id": "5a72db0b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -665,7 +665,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "09c1a43a",
+   "id": "01335c40",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -680,7 +680,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7371a75e",
+   "id": "d86d3dcc",
    "metadata": {
     "tags": [
      "solution"
@@ -697,7 +697,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e254f618",
+   "id": "186f40a5",
    "metadata": {
     "tags": []
    },
@@ -712,7 +712,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9817ee9d",
+   "id": "af4cf127",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -729,7 +729,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "68709b9f",
+   "id": "1f145891",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": [
@@ -743,7 +743,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5c1b84fa",
+   "id": "c2a072a4",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -754,7 +754,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "35a1b628",
+   "id": "f7bb217b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -764,7 +764,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0741cd51",
+   "id": "dfdd1272",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -782,7 +782,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9ea04106",
+   "id": "7ddd581a",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -794,7 +794,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "df0aa4e4",
+   "id": "2f7fe43c",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -813,7 +813,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "888fdbee",
+   "id": "f01095a7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -822,7 +822,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "3c6e2c10",
+   "id": "bad091aa",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -838,7 +838,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c959fb2f",
+   "id": "e49c55e0",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -847,7 +847,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e7249572",
+   "id": "af990b9a",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -859,7 +859,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "17b8c4b3",
+   "id": "291f2062",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -872,7 +872,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "fedb48d8",
+   "id": "919aea73",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -886,7 +886,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2ddc0100",
+   "id": "26f2a9c1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -898,7 +898,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4d4d2810",
+   "id": "54780ac3",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -918,7 +918,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "432f2f53",
+   "id": "e7187b29",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -942,7 +942,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9d6397cf",
+   "id": "78521f46",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -954,7 +954,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "779bcf13",
+   "id": "036e6086",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -976,7 +976,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "752616b6",
+   "id": "3b756b7a",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -988,7 +988,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "19d01864",
+   "id": "8256f4b5",
    "metadata": {
     "lines_to_next_cell": 2,
     "tags": [
@@ -1058,7 +1058,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a7cdf035",
+   "id": "78403aaa",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1070,7 +1070,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4bd99022",
+   "id": "9bb6c797",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1086,7 +1086,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2ca0425e",
+   "id": "c32cbd8b",
    "metadata": {
     "tags": []
    },
@@ -1101,7 +1101,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8ed754f6",
+   "id": "4c86dd42",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1113,7 +1113,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c90cd932",
+   "id": "8f728ded",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1135,7 +1135,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "23d27b74",
+   "id": "028d3bbc",
    "metadata": {
     "tags": []
    },
@@ -1151,7 +1151,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0afeb5a4",
+   "id": "b63aac86",
    "metadata": {
     "tags": []
    },
@@ -1161,7 +1161,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "61e418f1",
+   "id": "b0ad5935",
    "metadata": {
     "tags": []
    },
@@ -1178,7 +1178,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7b6c2411",
+   "id": "0cf2b9a5",
    "metadata": {
     "title": "Loading the test dataset"
    },
@@ -1198,7 +1198,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "26cd8983",
+   "id": "35cc9b35",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1210,7 +1210,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f1f5cc1e",
+   "id": "4146969c",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1223,7 +1223,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f3cd6599",
+   "id": "f99f676c",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1233,7 +1233,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "cefcba31",
+   "id": "a7450339",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1251,7 +1251,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7f0d387c",
+   "id": "f2bfd025",
    "metadata": {
     "tags": [
      "solution"
@@ -1288,7 +1288,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "bfda4b7f",
+   "id": "20af6915",
    "metadata": {
     "lines_to_next_cell": 0,
     "tags": []
@@ -1300,7 +1300,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a5b2f903",
+   "id": "4543762f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1313,7 +1313,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "935009d9",
+   "id": "d0b8a2ec",
    "metadata": {
     "tags": []
    },
@@ -1328,7 +1328,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7683e70e",
+   "id": "61e58d7f",
    "metadata": {
     "tags": []
    },
@@ -1339,7 +1339,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3532d6ed",
+   "id": "b8dc4640",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1353,7 +1353,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c8c90e6b",
+   "id": "ed9b7104",
    "metadata": {
     "tags": []
    },
@@ -1368,7 +1368,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c4c3eff4",
+   "id": "0add70ab",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1383,7 +1383,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1eff97c0",
+   "id": "274d8226",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1404,7 +1404,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c1281384",
+   "id": "b6b54bbe",
    "metadata": {
     "title": "Another visualization function"
    },
@@ -1433,7 +1433,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "42cf09ad",
+   "id": "979ca20f",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1449,7 +1449,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "30ea4f96",
+   "id": "b779b535",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1465,7 +1465,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6768c3d0",
+   "id": "c44f0cf1",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1480,7 +1480,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0287a687",
+   "id": "f402bb74",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1494,7 +1494,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6459abb5",
+   "id": "12542a6d",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1514,7 +1514,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ca40f5a7",
+   "id": "8d3e6d86",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1550,7 +1550,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "94276a21",
+   "id": "ca1d1242",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1564,7 +1564,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a7597f89",
+   "id": "6f530e82",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1580,7 +1580,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9cb53f3f",
+   "id": "5904b20b",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1596,7 +1596,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6a43a957",
+   "id": "72ca6d87",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1619,7 +1619,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "50e69413",
+   "id": "a0baa037",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-info\"><h3>Task 5.1: Explore the style space</h3>\n",
@@ -1631,7 +1631,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c71b768e",
+   "id": "bc0062db",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1666,12 +1666,12 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a4676c24",
+   "id": "ba428131",
    "metadata": {
     "lines_to_next_cell": 0
    },
    "source": [
-    "<div class=\"alert alert-block alert-info\"><h3>Task 5.1: Adding color to the style space</h3>\n",
+    "<div class=\"alert alert-block alert-info\"><h3>Task 5.2: Adding color to the style space</h3>\n",
     "We know that color is important. Does interpreting the style space as colors help us understand better?\n",
     "\n",
     "Let's use the style space to color the PCA plot.\n",
@@ -1682,7 +1682,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b5838e7c",
+   "id": "2ee2e061",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1709,7 +1709,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2aff4255",
+   "id": "a26903b8",
    "metadata": {
     "lines_to_next_cell": 0
    },
@@ -1723,12 +1723,12 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a80963e3",
+   "id": "7f72dfb5",
    "metadata": {
     "lines_to_next_cell": 0
    },
    "source": [
-    "<div class=\"alert alert-block alert-info\"><h3>Task 5.2: Using the images to color the style space</h3>\n",
+    "<div class=\"alert alert-block alert-info\"><h3>Task 5.3: Using the images to color the style space</h3>\n",
     "Finally, let's just use the colors from the images themselves!\n",
     "The maximum value in the image (since they are \"black-and-color\") can be used as a color!\n",
     "\n",
@@ -1740,7 +1740,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "65e070c3",
+   "id": "be62a09b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1762,7 +1762,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6f5d93bb",
+   "id": "f16385e4",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-warning\"><h3>Questions</h3>\n",
@@ -1774,7 +1774,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1c2bf24d",
+   "id": "e0f59eee",
    "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\"><h2>Checkpoint 5</h2>\n",
@@ -1792,7 +1792,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c5cb1ade",
+   "id": "0bbc6cd2",
    "metadata": {},
    "source": [
     "# Bonus!\n",