diff --git a/.gitignore b/.gitignore
index 2f3aafd..c08417d 100755
--- a/.gitignore
+++ b/.gitignore
@@ -3,6 +3,7 @@
 notebooks/data/
 notebooks/jax_nb_data/
 notebooks/torch_nb_data/
+hssm_paper/
 
 notebooks/test_notebooks/data/
 
diff --git a/docs/basic_tutorial/basic_tutorial_lan_jax.ipynb b/docs/basic_tutorial/basic_tutorial_lan_jax.ipynb
index bdb000d..883b94f 100644
--- a/docs/basic_tutorial/basic_tutorial_lan_jax.ipynb
+++ b/docs/basic_tutorial/basic_tutorial_lan_jax.ipynb
@@ -263,13 +263,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
     "# LOAD NETWORK\n",
     "# Test properties of network\n",
-    "jax_net = lanfactory.trainers.MLPJaxFactory(network_config=network_config, train=True)\n",
+    "jax_net = lanfactory.trainers.MLPJaxFactory(network_config=network_config,\n",
+    "                                            train=True)\n",
     "\n",
     "# Save model config\n",
     "# model_folder = os.path.join(\"data\", \"jax_models\", MODEL)\n",