diff --git a/cs2109s/labs/flake.nix b/cs2109s/labs/flake.nix
index 9778b27..4c8e816 100644
--- a/cs2109s/labs/flake.nix
+++ b/cs2109s/labs/flake.nix
@@ -2,7 +2,7 @@
   description = "A very basic flake";
 
   inputs = {
-    nixpkgs.url = "github:nixos/nixpkgs/nixos-23.11";
+    nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
     flake-utils.url = "github:numtide/flake-utils";
   };
 
@@ -11,26 +11,31 @@
       inherit system;
       config = {
         allowUnfree = true;
-	cudaSupport = true;
       };
-    };
-    in rec {
+    }; in rec {
+      overlays = [(final: prev: {
+        pythonPackagesExtensions = [(py-final: py-prev: {
+          torch = py-final.torch-bin;
+          torchvision = py-final.torchvision-bin;
+          torchaudio = py-final.torchaudio-bin;
+        })];
+      })];
       devShell = pkgs.mkShell {
          buildInputs = with pkgs; [
-	 cudaPackages.cudatoolkit
-          (python3.withPackages(ps: with ps; [
+          (python3.withPackages(ps: with ps; 
+          [
             ipython
             jupyter
             numpy
             pandas
-	    matplotlib
-	    torch
-	    scikit-learn
-	    timeout-decorator
-	    torchvision
+            matplotlib
+            torch
+            scikit-learn
+            timeout-decorator
+            torchvision
+            seaborn
           ]))
         ];
-        shellHook = "jupyter notebook --ip='100.64.0.8'";
       };
     }
   );