diff --git a/04_DenoiSplit/solution.ipynb b/04_DenoiSplit/solution.ipynb
index 617b299..0b305a3 100755
--- a/04_DenoiSplit/solution.ipynb
+++ b/04_DenoiSplit/solution.ipynb
@@ -40,7 +40,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "b6e41f1d",
    "metadata": {},
    "outputs": [],
@@ -63,7 +63,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "47dbd8fb",
    "metadata": {
     "lines_to_next_cell": 0
@@ -79,14 +79,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 11,
    "id": "84e96ca6",
    "metadata": {},
    "outputs": [],
    "source": [
     "import sys\n",
     "\n",
-    "sys.path.append(\"../../\")\n",
+    "sys.path.append(\"../denoiSplit\")\n",
     "\n",
     "from pytorch_lightning.loggers import TensorBoardLogger\n",
     "from pytorch_lightning.callbacks import ModelCheckpoint\n",
diff --git a/setup.sh b/setup.sh
index 252b14c..998d780 100755
--- a/setup.sh
+++ b/setup.sh
@@ -16,7 +16,7 @@ fi
 if [[ "$CONDA_DEFAULT_ENV" == "$ENV" ]]; then
     conda install -y pytorch-gpu cuda-toolkit=11.8 torchvision -c nvidia -c conda-forge -c pytorch
     #mamba install -y pytorch torchvision pytorch-cuda=11.8 -c pytorch -c nvidia
-    pip install jupytext black nbconvert scikit-learn ipykernel gdown "careamics[examples,tensorboard] @ git+https://github.com/CAREamics/careamics.git"
+    pip install jupytext black nbconvert scikit-learn ipykernel gdown "careamics[examples,tensorboard] albumentations wandb ml_collections @ git+https://github.com/CAREamics/careamics.git"
     python -m ipykernel install --user --name "05_image_restoration"
 fi
 
@@ -35,3 +35,4 @@ cd ../
 mkdir ER/
 cd ER/
 gdown 1Bho6Oymfxi7OV0tPb9wkINkVOCpTaL7M
+cd ../..
\ No newline at end of file