diff --git a/devShells/uniPython/flake.nix b/devShells/uniPython/flake.nix
index c39ecbb..71c8bac 100644
--- a/devShells/uniPython/flake.nix
+++ b/devShells/uniPython/flake.nix
@@ -42,6 +42,12 @@
           ];
 
           MPLBACKEND = "kitcat";
+          PYTHONSTARTUP = "${pkgs.writeText "pythonstartup.py" ''
+            import numpy as np
+            import pandas as pd
+            import torch
+            from torch import nn
+            ''}";
         };
       });
     };