diff --git a/program/cohort.ipynb b/program/cohort.ipynb
index 4b197ec..3162409 100644
--- a/program/cohort.ipynb
+++ b/program/cohort.ipynb
@@ -1868,12 +1868,13 @@
     "import json\n",
     "import os\n",
     "import tarfile\n",
+    "\n",
     "from pathlib import Path\n",
+    "from comet_ml import Experiment\n",
     "\n",
     "import keras\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "from comet_ml import Experiment\n",
     "from keras import Input\n",
     "from keras.layers import Dense\n",
     "from keras.models import Sequential\n",
diff --git a/requirements.txt b/requirements.txt
index d7b2923..8a903cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 numpy
 pandas
 matplotlib
+comet-ml
 scikit-learn
 python-dotenv
 ipytest
@@ -13,4 +14,3 @@ nbdev
 nbclient
 ipykernel
 Flask
-comet-ml