diff --git a/examples/01_video_embeddings_tsne_plot.ipynb b/examples/01_video_embeddings_tsne_plot.ipynb
index d622bbb..80e64dd 100644
--- a/examples/01_video_embeddings_tsne_plot.ipynb
+++ b/examples/01_video_embeddings_tsne_plot.ipynb
@@ -38,7 +38,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "id": "41d7b1c6-157e-4c28-b362-2119779c7909",
    "metadata": {},
    "outputs": [
@@ -55,14 +55,14 @@
     }
    ],
    "source": [
-    "from feluda import Feluda\n",
-    "from feluda.models.media_factory import VideoFactory\n",
+    "# from feluda import Feluda\n",
+    "# from feluda.models.media_factory import VideoFactory\n",
     "\n",
-    "from huggingface_hub import hf_hub_download\n",
-    "import cv2\n",
-    "import tarfile\n",
-    "import matplotlib.pyplot as plt\n",
-    "from matplotlib.offsetbox import OffsetImage, AnnotationBbox"
+    "# from huggingface_hub import hf_hub_download\n",
+    "# import cv2\n",
+    "# import tarfile\n",
+    "# import matplotlib.pyplot as plt\n",
+    "# from matplotlib.offsetbox import OffsetImage, AnnotationBbox"
    ]
   },
   {
diff --git a/tests/feluda_unit_tests/models/test_media_factory.py b/tests/feluda_unit_tests/models/test_media_factory.py
index 30838dc..2373b2c 100644
--- a/tests/feluda_unit_tests/models/test_media_factory.py
+++ b/tests/feluda_unit_tests/models/test_media_factory.py
@@ -6,6 +6,7 @@
 import tempfile
 from feluda.models.media_factory import ImageFactory, VideoFactory, AudioFactory
 
+
 class Test(unittest.TestCase):
     # Class variables to store file paths for cleanup
     test_files = []
@@ -19,7 +20,7 @@ def setUpClass(cls):
         cls.expected_files = [
             "text-in-image-test-hindi.png",
             "cat_vid_2mb.mp4.mp4",
-            "audio.wav.wav"
+            "audio.wav.wav",
         ]
 
     @classmethod