diff --git a/.gitignore b/.gitignore
index c85460c..15f54f5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -23,8 +23,5 @@
 *.nav
 *.snm
 *.mp4
-*.png
-*.jpg
-*.jpeg
 *.npy
 
diff --git a/Deep-SAD-PyTorch/devenv.lock b/Deep-SAD-PyTorch/devenv.lock
new file mode 100644
index 0000000..a1a881e
--- /dev/null
+++ b/Deep-SAD-PyTorch/devenv.lock
@@ -0,0 +1,103 @@
+{
+  "nodes": {
+    "devenv": {
+      "locked": {
+        "dir": "src/modules",
+        "lastModified": 1754730435,
+        "owner": "cachix",
+        "repo": "devenv",
+        "rev": "d1388a093a7225c2abe8c244109c5a4490de4077",
+        "type": "github"
+      },
+      "original": {
+        "dir": "src/modules",
+        "owner": "cachix",
+        "repo": "devenv",
+        "type": "github"
+      }
+    },
+    "flake-compat": {
+      "flake": false,
+      "locked": {
+        "lastModified": 1747046372,
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "rev": "9100a0f413b0c601e0533d1d94ffd501ce2e7885",
+        "type": "github"
+      },
+      "original": {
+        "owner": "edolstra",
+        "repo": "flake-compat",
+        "type": "github"
+      }
+    },
+    "git-hooks": {
+      "inputs": {
+        "flake-compat": "flake-compat",
+        "gitignore": "gitignore",
+        "nixpkgs": [
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1754416808,
+        "owner": "cachix",
+        "repo": "git-hooks.nix",
+        "rev": "9c52372878df6911f9afc1e2a1391f55e4dfc864",
+        "type": "github"
+      },
+      "original": {
+        "owner": "cachix",
+        "repo": "git-hooks.nix",
+        "type": "github"
+      }
+    },
+    "gitignore": {
+      "inputs": {
+        "nixpkgs": [
+          "git-hooks",
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1709087332,
+        "owner": "hercules-ci",
+        "repo": "gitignore.nix",
+        "rev": "637db329424fd7e46cf4185293b9cc8c88c95394",
+        "type": "github"
+      },
+      "original": {
+        "owner": "hercules-ci",
+        "repo": "gitignore.nix",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1754299112,
+        "owner": "cachix",
+        "repo": "devenv-nixpkgs",
+        "rev": "16c21c9f5c6fb978466e91182a248dd8ca1112ac",
+        "type": "github"
+      },
+      "original": {
+        "owner": "cachix",
+        "ref": "rolling",
+        "repo": "devenv-nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "devenv": "devenv",
+        "git-hooks": "git-hooks",
+        "nixpkgs": "nixpkgs",
+        "pre-commit-hooks": [
+          "git-hooks"
+        ]
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}
diff --git a/Deep-SAD-PyTorch/pyproject.toml b/Deep-SAD-PyTorch/pyproject.toml
index 8c8a5ca..508c8f9 100644
--- a/Deep-SAD-PyTorch/pyproject.toml
+++ b/Deep-SAD-PyTorch/pyproject.toml
@@ -12,6 +12,8 @@ dependencies = [
     "kiwisolver>=1.4.8",
     "matplotlib>=3.10.3",
     "numpy>=2.3.1",
+    "onnx>=1.18.0",
+    "onnxscript>=0.3.2",
     "pandas>=2.3.0",
     "pillow>=11.2.1",
     "pyparsing>=3.2.3",
@@ -23,6 +25,7 @@ dependencies = [
     "six>=1.17.0",
     "torch-receptive-field",
     "torchscan>=0.1.1",
+    "visualtorch>=0.2.4",
 ]
 
 [tool.uv.sources]
diff --git a/thesis/flake.nix b/thesis/flake.nix
index e20016e..26a512a 100644
--- a/thesis/flake.nix
+++ b/thesis/flake.nix
@@ -25,6 +25,7 @@
           texlab
           zathura
           wmctrl
+          python312
         ];
       in
       {