diff --git a/flake.nix b/flake.nix
index 6b0d8b1..f2911fd 100644
--- a/flake.nix
+++ b/flake.nix
@@ -30,14 +30,27 @@
             };
 
             dontUnpack = true;
-            nativeBuildInputs = [ python3 ];
+            nativeBuildInputs = [ python3 gnused ];
 
             buildPhase = ''
               runHook preBuild
 
-              cp $gutenbergVirginian book1.txt
-              cp $gutenbergCattleDiseases book2.txt
-              cp $gutenbergVetAnatomy book3.txt
+              head -n14969 $gutenbergVirginian \
+                | sed '1,136d' \
+                > book1.txt
+              head -n25236 $gutenbergCattleDiseases \
+                | sed '1,282d' \
+                | sed 's/_//g' \
+                | sed 's/=//g' \
+                | sed -E '/^(\s+\*){5}/,/^(\s+\*){5}/d' \
+                | sed '/^\s*\[Illustration:/,/^\]/d' \
+                > book2.txt
+              head -n43690 $gutenbergVetAnatomy \
+                | sed '1,367d' \
+                | sed '/^\[Illustration:/,/^\]/d' \
+                | sed 's/_//g' \
+                | sed 's/=//g' \
+                > book3.txt
 
               python3 $src
 
@@ -84,15 +97,19 @@
           };
         container = with pkgs;
           dockerTools.buildImage {
-            name = "babbler";
+            name = "quay.io/millironx/babbler";
             tag = "latest";
             copyToRoot = [ babbler ];
-            config = { Cmd = [ "${babbler}/bin/babble" ]; };
+            config = {
+              Cmd = [ "${babbler}/bin/babble" ];
+              WorkingDir = "${babbler}/bin";
+            };
           };
       in {
         packages = {
           inherit babbler container;
           default = babbler;
         };
+        devShells.default = with pkgs; mkShell { inputsFrom = [ babbler ]; };
       });
 }