diff --git a/assets/tcmi/logo-mpg.png b/assets/tcmi/logo_MPG.png
similarity index 100%
rename from assets/tcmi/logo-mpg.png
rename to assets/tcmi/logo_MPG.png
diff --git a/assets/tcmi/logo-nomad.png b/assets/tcmi/logo_NOMAD.png
similarity index 100%
rename from assets/tcmi/logo-nomad.png
rename to assets/tcmi/logo_NOMAD.png
diff --git a/tcmi.ipynb b/tcmi.ipynb
index 9fa45cbd82f46a9ff9c8e3dc791a8764b4452f88..9cf2bed6960e645fe8b67e569dd798df842a23a3 100644
--- a/tcmi.ipynb
+++ b/tcmi.ipynb
@@ -6,34 +6,15 @@
     "pycharm": {}
    },
    "source": [
-    "<div id=\"teaser\" style=' background-position:  right center; background-size: 00px; background-repeat: no-repeat; \n",
-    "    padding-top: 20px;\n",
-    "    padding-right: 10px;\n",
-    "    padding-bottom: 170px;\n",
-    "    padding-left: 10px;\n",
-    "    border-bottom: 14px double #333;\n",
-    "    border-top: 14px double #333;' > \n",
-    "\n",
-    "   \n",
-    "   <div style=\"text-align:center\">\n",
-    "    <b><font size=\"6.4\">Introduction to total cumulative mutual information</font></b>    \n",
-    "  </div>\n",
-    "    \n",
-    "<p>\n",
-    " created by:\n",
-    " Benjamin Regler<sup>1</sup>, \n",
-    " Matthias Scheffler<sup>1</sup>,\n",
-    " and Luca M. Ghiringhelli<sup> 1</sup> <br><br>\n",
-    "<sup>1</sup> Fritz Haber Institute of the Max Planck Society, Faradayweg 4-6, D-14195 Berlin, Germany <br>\n",
-    "<span class=\"nomad--last-updated\" data-version=\"v1.0.2\">[Last updated: January 20, 2020]</span>\n",
-    "</p>\n",
-    "  \n",
-    "<div> \n",
-    "    <img style=\"float: left;\" src=\"assets/tcmi/logo-mpg.png\" width=\"200\"> \n",
-    "    <img style=\"float: right;\" src=\"assets/tcmi/logo-nomad.png\" width=\"250\">\n",
-    "</div>\n",
-    "</p>\n",
-    "</div>"
+    "<img  src=\"assets/tcmi/header.jpg\" width=\"900\"> "
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<img  style=\"float: left;\" src=\"assets/tcmi/logo_NOMAD.png\" width=300>\n",
+    "<img  style=\"float: right;\" src=\"assets/tcmi/logo_MPG.png\" width=170> "
    ]
   },
   {
@@ -72,10 +53,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {
     "ExecuteTime": {
-     "start_time": "2020-02-06T20:03:16.271Z"
+     "end_time": "2021-11-11T17:36:57.497040Z",
+     "start_time": "2021-11-11T17:36:57.488774Z"
     },
     "init_cell": true
    },
@@ -96,15 +78,25 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {
     "ExecuteTime": {
-     "start_time": "2020-02-06T20:03:16.273Z"
+     "end_time": "2021-11-11T17:36:58.898232Z",
+     "start_time": "2021-11-11T17:36:57.498482Z"
     },
     "hidden": true,
     "init_cell": true
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/sbailo/anaconda3/envs/ai_toolkit/lib/python3.7/site-packages/sklearn/utils/deprecation.py:143: FutureWarning: The sklearn.metrics.scorer module is  deprecated in version 0.22 and will be removed in version 0.24. The corresponding classes / functions should instead be imported from sklearn.metrics. Anything that cannot be imported from sklearn.metrics is now part of the private API.\n",
+      "  warnings.warn(message, FutureWarning)\n"
+     ]
+    }
+   ],
    "source": [
     "import os\n",
     "import re\n",