diff --git a/python-data-science.nix b/python-data-science.nix
index 345cf04..878ec1d 100644
--- a/python-data-science.nix
+++ b/python-data-science.nix
@@ -1,8 +1,8 @@
-{ config, lib, pkgs, ... }:
+{ pkgs ? import <nixpkgs> { } }:
 
 with pkgs;
 
-{
+mkShell {
   buildInputs = [
     python3
     python3Packages.jupyter
@@ -12,4 +12,10 @@ with pkgs;
     python3Packages.matplotlib
     python3Packages.seaborn
   ];
+
+  shellHook = ''
+    trap "kill 0" EXIT
+
+    jupyter notebook
+  '';
 }