diff --git a/python-data-science.nix b/python-data-science.nix
index 345cf04..fb3cc62 100644
--- a/python-data-science.nix
+++ b/python-data-science.nix
@@ -2,7 +2,7 @@
 
 with pkgs;
 
-{
+mkShell {
   buildInputs = [
     python3
     python3Packages.jupyter
@@ -12,4 +12,10 @@ with pkgs;
     python3Packages.matplotlib
     python3Packages.seaborn
   ];
+
+  shellHook = ''
+    trap "kill 0" EXIT
+
+    jupyter notebook
+  '';
 }