diff --git a/flake.nix b/flake.nix
index 39fe9f9..7bf72ea 100644
--- a/flake.nix
+++ b/flake.nix
@@ -81,12 +81,20 @@
         '';
       };
 
+      pythonEnv = pkgs.python3.withPackages (
+        ps: with ps; [
+          pandas
+          matplotlib
+        ]
+      );
+
     in
     {
       packages.${system} = {
         default = misth;
         misth = misth;
         mruby = mruby;
+        pythonEnv = pythonEnv;
       };
 
       devShells.${system}.default = pkgs.mkShell {
@@ -104,6 +112,9 @@
           sdl3
           sdl3-ttf
           sdl3-image
+
+          pythonEnv
+          ruff
         ];
 
         shellHook = ''
diff --git a/tests/grapher.py b/tests/grapher.py
index 4f7d732..36bf0b7 100644
--- a/tests/grapher.py
+++ b/tests/grapher.py
@@ -21,4 +21,4 @@ plt.legend()
 
 plt.show()
 
-plt.savefig("benchmark.png", dpi=200, bbox_inches="tight")
\ No newline at end of file
+plt.savefig("benchmark.png", dpi=200, bbox_inches="tight")