diff --git a/alex/personal.nix b/alex/personal.nix
index 529532a..2ff953a 100644
--- a/alex/personal.nix
+++ b/alex/personal.nix
@@ -26,9 +26,19 @@ in
     shotwell
     nextcloud-client
     bitwarden-desktop
-    signal-desktop
+    (symlinkJoin {
+      name = "signal-desktop-wrapped";
+      paths = [ signal-desktop ];
+      nativeBuildInputs = [ makeWrapper ];
+      postBuild = ''
+        rm $out/bin/signal-desktop
+        makeWrapper ${signal-desktop}/bin/signal-desktop $out/bin/signal-desktop \
+          --add-flags "--password-store=gnome-libsecret"
+      '';
+    })
     spotify
-    gemini-cli
+    unstable.gemini-cli
+    unstable.claude-code
 
     foliate
 
diff --git a/alex/sway.nix b/alex/sway.nix
index fc7b31e..c0ab550 100644
--- a/alex/sway.nix
+++ b/alex/sway.nix
@@ -171,7 +171,9 @@ in
 
         "${modifier}+Shift+l" = "exec swaylock -f";
 
-        "${modifier}+Shift+s" = "exec grim -g \"$(slurp)\" - | wl-copy";
+        "${modifier}+Shift+s" = "exec grim -g \"$(slurp)\" - | tee \"/home/alex/Pictures/Screenshots/$(date +'%Y%m%d_%Hh%Mm%Ss_grim.png')\" | wl-copy";
+
+        "Print" = "exec grim - | tee \"/home/alex/Pictures/Screenshots/$(date +'%Y%m%d_%Hh%Mm%Ss_grim.png')\" | wl-copy";
 
         # Workspaces
         "--whole-window ${modifier}+button4" = "workspace prev";
diff --git a/flake.lock b/flake.lock
index 0522ac9..360504f 100644
--- a/flake.lock
+++ b/flake.lock
@@ -92,11 +92,11 @@
     },
     "nixpkgs_2": {
       "locked": {
-        "lastModified": 1773222311,
-        "narHash": "sha256-BHoB/XpbqoZkVYZCfXJXfkR+GXFqwb/4zbWnOr2cRcU=",
+        "lastModified": 1773375660,
+        "narHash": "sha256-SEzUWw2Rf5Ki3bcM26nSKgbeoqi2uYy8IHVBqOKjX3w=",
         "owner": "NixOS",
         "repo": "nixpkgs",
-        "rev": "0590cd39f728e129122770c029970378a79d076a",
+        "rev": "3e20095fe3c6cbb1ddcef89b26969a69a1570776",
         "type": "github"
       },
       "original": {
diff --git a/hosts/count/configuration.nix b/hosts/count/configuration.nix
index 6bb804f..856d2a4 100644
--- a/hosts/count/configuration.nix
+++ b/hosts/count/configuration.nix
@@ -24,6 +24,27 @@
   # Use latest kernel.
   boot.kernelPackages = pkgs.linuxPackages_latest;
 
+# Enable AMD GPU drivers
+	boot.initrd.kernelModules = [ "amdgpu" ];
+	services.xserver.videoDrivers = [ "amdgpu" ];
+
+	# Enable ROCm support
+	hardware.graphics.enable = true;
+	hardware.amdgpu.opencl.enable = true;
+
+	# Add the ROCm packages to your system
+	environment.systemPackages = with pkgs; [
+		rocmPackages.rocminfo
+		rocmPackages.rocm-smi
+	];
+
+	services.ollama = {
+		enable = true;
+		acceleration = "rocm";
+		# Force the RX 6600 to use compatible ROCm kernels
+		rocmOverrideGfx = "10.3.0";
+	};
+
   # Setup keyfile
   boot.initrd.secrets = {
     "/boot/crypto_keyfile.bin" = null;