diff --git a/hosts/workstation/ollama.nix b/hosts/workstation/ollama.nix
index 2afe3af..487c882 100644
--- a/hosts/workstation/ollama.nix
+++ b/hosts/workstation/ollama.nix
@@ -9,11 +9,12 @@ with lib; let
 in {
   options.ollama = {enable = mkEnableOption "is ollama host";};
   config = mkIf cfg.enable {
+    environment.systemPackages = [ pkgs.vscode ];
     services.ollama = {
       enable = true;
       package = pkgs.ollama-rocm;
       # Optional: preload models, see https://ollama.com/library
-      loadModels = ["qwen2.5-coder:1.5b"];
+      loadModels = ["qwen2.5-coder" "deepseek-r1"];
     };
   };
 }