diff --git a/example/paper_example_nsf.ipynb b/example/paper_example_nsf.ipynb
index e693286..d512e3e 100644
--- a/example/paper_example_nsf.ipynb
+++ b/example/paper_example_nsf.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "37dcaedc",
+   "id": "94003a1d",
    "metadata": {},
    "source": [
     "# Neural Spline Flow on a Circular and a Normal Coordinate"
@@ -11,7 +11,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0ba0ffb9",
+   "id": "df035970",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -31,7 +31,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "15008e69",
+   "id": "1d04fa87",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -51,7 +51,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2c113dfe",
+   "id": "7bda37d4",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -61,7 +61,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "dc02e62b",
+   "id": "c73b0c95",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -84,7 +84,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ab4234be",
+   "id": "3a22921b",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -109,7 +109,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "be7a29e8",
+   "id": "3eba01f9",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -127,7 +127,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "34efecc6",
+   "id": "156b5907",
    "metadata": {
     "scrolled": false
    },
@@ -180,7 +180,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c4a4b07d",
+   "id": "95fe8e35",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -204,24 +204,6 @@
     "\n",
     "plt.show()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "e08d4f90",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "bool(0)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a359e9c7",
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {