diff --git a/flake.nix b/flake.nix
index d9bece6..0348feb 100644
--- a/flake.nix
+++ b/flake.nix
@@ -40,7 +40,7 @@
           };
         };
 
-        python-path = pkgs.python3Packages.buildPythonPackage rec {
+        ultralytics = pkgs.python3Packages.buildPythonPackage rec {
           # pname = "opencv-python";
           # version = "4.10.0.84";
 
@@ -64,6 +64,8 @@
             pythonPackages.torchvision
             pythonPackages.pandas
             pythonPackages.matplotlib
+            pythonPackages.seaborn
+            ultralytics-thop
 
           ];
 
@@ -97,7 +99,7 @@
           };
         };
 
-        env = pkgs.python312.withPackages (ps: [ ultralytics-thop ]);
+        env = pkgs.python312.withPackages (ps: [ ultralytics ]);
 
       in {
         packages = {