diff --git a/docs/examples/JOSS/JOSS_Example.py b/docs/examples/JOSS/JOSS_Example.py
index bae8f0d..7a7f70e 100644
--- a/docs/examples/JOSS/JOSS_Example.py
+++ b/docs/examples/JOSS/JOSS_Example.py
@@ -87,4 +87,4 @@
 save = True
 if save:
     plt.tight_layout(pad=0.0, h_pad=0, w_pad=0)
-    plt.savefig('JOSS.png')
+    plt.savefig('JOSS.pdf')
diff --git a/docs/examples/nns/mnist.ipynb b/docs/examples/nns/mnist.ipynb
index 036d653..bd80d6d 100644
--- a/docs/examples/nns/mnist.ipynb
+++ b/docs/examples/nns/mnist.ipynb
@@ -14,7 +14,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "1b37649d-ae72-49dc-855e-5109b13d9433",
    "metadata": {},
    "outputs": [],
@@ -43,7 +43,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "97fe038d-11a6-4114-bd6a-a09e83a96c7c",
    "metadata": {},
    "outputs": [],
@@ -68,7 +68,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
    "id": "165cef19-2b94-4490-a007-714df29ba5c1",
    "metadata": {},
    "outputs": [],
@@ -88,7 +88,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 4,
    "id": "34def998-0a7e-43f9-bc9c-d890f5a0b7d5",
    "metadata": {},
    "outputs": [],
@@ -117,7 +117,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 5,
    "id": "705a76af-42bb-4ca6-9f8c-f1f6d9dbad6a",
    "metadata": {},
    "outputs": [],
@@ -147,7 +147,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 6,
    "id": "cac7a15f-8445-4eb3-ad00-bfdd15e201d8",
    "metadata": {},
    "outputs": [],
@@ -189,7 +189,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 7,
    "id": "b40c7cf2-4295-4bf3-8328-e47ba7ba16d5",
    "metadata": {},
    "outputs": [],
@@ -225,7 +225,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "id": "9e32f6f7-7937-4ceb-a043-93eb8e85f9b1",
    "metadata": {},
    "outputs": [],
@@ -258,13 +258,60 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "id": "7ed4c926-f239-42b0-93c2-fcf48171adaa",
    "metadata": {
     "scrolled": true,
     "tags": []
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "------------------------------\n",
+      "Epoch: 1\n",
+      "Accuracy: 0.5325999855995178\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 2\n",
+      "Accuracy: 0.6539000272750854\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 3\n",
+      "Accuracy: 0.6708999872207642\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 4\n",
+      "Accuracy: 0.7269999980926514\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 5\n",
+      "Accuracy: 0.737500011920929\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 6\n",
+      "Accuracy: 0.7563999891281128\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 7\n",
+      "Accuracy: 0.7552000284194946\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 8\n",
+      "Accuracy: 0.784500002861023\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 9\n",
+      "Accuracy: 0.7868000268936157\n",
+      "------------------------------\n",
+      "------------------------------\n",
+      "Epoch: 10\n",
+      "Accuracy: 0.779699981212616\n",
+      "------------------------------\n"
+     ]
+    }
+   ],
    "source": [
     "e = 0\n",
     "while f.epochs < 10:\n",
diff --git a/pyproject.toml b/pyproject.toml
index f1b60e9..b24d023 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name="cbx"
-version="0.1.5"
+version="0.1.6"
 authors = [
     {name = "Tim Roith", email = "tim.roith@desy.de"},
 ]