diff --git a/hosts/pinwheel/home.nix b/hosts/pinwheel/home.nix
index 43e4c41..3981e0d 100644
--- a/hosts/pinwheel/home.nix
+++ b/hosts/pinwheel/home.nix
@@ -27,7 +27,6 @@
           pkgs.onlyoffice-desktopeditors
           pkgs.wdisplays
           pkgs.vlc
-          pkgs.claude-code
           pkgs.opencode
         ];
 
diff --git a/hosts/pinwheel/modules/ai/default.nix b/hosts/pinwheel/modules/ai/default.nix
new file mode 100644
index 0000000..a91e044
--- /dev/null
+++ b/hosts/pinwheel/modules/ai/default.nix
@@ -0,0 +1,29 @@
+{
+  pkgs,
+  lib,
+  config,
+  ...
+}:
+let
+  enabled = config.mod.ai.enable;
+in
+{
+  options = {
+    mod.ai = {
+      enable = lib.mkEnableOption "enable ai module";
+    };
+  };
+
+  config = lib.mkIf enabled {
+    home-manager.users.alex = {
+      home.packages = [
+        pkgs.claude-code
+      ];
+
+      programs.zsh.shellAliases = {
+        wclaude = "CLAUDE_CONFIG_DIR=$HOME/.claude-personal claude";
+        pclaude = "CLAUDE_CONFIG_DIR=$HOME/.claude-work claude";
+      };
+    };
+  };
+}
diff --git a/hosts/pinwheel/modules/default.nix b/hosts/pinwheel/modules/default.nix
index 8f43ffa..5a34294 100644
--- a/hosts/pinwheel/modules/default.nix
+++ b/hosts/pinwheel/modules/default.nix
@@ -42,6 +42,8 @@ in
       };
       vm.enable = true;
       scripts.enable = true;
+
+      ai.enable = true;
     };
   };
 }