diff --git a/flake.nix b/flake.nix
index 3dbb079..da7c8ed 100644
--- a/flake.nix
+++ b/flake.nix
@@ -95,10 +95,13 @@
                   numpy
                   opencv4
                   boto3
+                  tensorboardx
+                  albumentations
+                  simplejson
                 ];
               in
               pkgs.mkShell {
-                packages = [ (pkgs.python3.withPackages perception-packages) ];
+                packages = [ pkgs.s3cmd pkgs.libGL ] ++ [ (pkgs.python3.withPackages perception-packages) ];
               };
           }; 
         }