diff --git a/common/.config/nvim/lua/plugins/ai.lua b/common/.config/nvim/lua/plugins/ai.lua
index 3c9dcdf..51e1b71 100644
--- a/common/.config/nvim/lua/plugins/ai.lua
+++ b/common/.config/nvim/lua/plugins/ai.lua
@@ -2,10 +2,10 @@ return {
     {
         "David-Kunz/gen.nvim",
         opts = {
-            model = "mistral", -- The default model to use.
-            host = "localhost", -- The host running the Ollama service.
+            model = "mistral:instruct", -- The default model to use.
+            host = "127.0.0.1", -- The host running the Ollama service.
             port = "11434", -- The port on which the Ollama service is listening.
-            display_mode = "float", -- The display mode. Can be "float" or "split".
+            display_mode = "split", -- The display mode. Can be "float" or "split".
             show_prompt = true, -- Shows the Prompt submitted to Ollama.
             show_model = true, -- Displays which model you are using at the beginning of your chat session.
             quit_map = "q", -- set keymap for quit
@@ -25,7 +25,8 @@ return {
             debug = false, -- Prints errors and the command which is run.
         },
         keys = {
-            { "<leader>]", "<cmd>Gen<cr>", desc = "AI Prompt", mode = { "n", "v" } },
+            { "<leader>[", ":Gen<cr>", desc = "AI Prompt", mode = { "n" } },
+            { "<leader>]", ":'<,'>Gen<cr>", desc = "Contextual AI Prompt", mode = { "v" } },
         },
     },
 }
diff --git a/common/.config/nvim/lua/plugins/code-debug.lua b/common/.config/nvim/lua/plugins/code-debug.lua
index ab4bf81..e6c8240 100644
--- a/common/.config/nvim/lua/plugins/code-debug.lua
+++ b/common/.config/nvim/lua/plugins/code-debug.lua
@@ -6,6 +6,7 @@ return {
     dependencies = {
         -- Creates a beautiful debugger UI
         "rcarriga/nvim-dap-ui",
+        "nvim-neotest/nvim-nio",
 
         -- Installs the debug adapters for you
         "williamboman/mason.nvim",
diff --git a/common/.config/nvim/lua/plugins/code-generic.lua b/common/.config/nvim/lua/plugins/code-generic.lua
index f0b0f10..084fda1 100644
--- a/common/.config/nvim/lua/plugins/code-generic.lua
+++ b/common/.config/nvim/lua/plugins/code-generic.lua
@@ -49,7 +49,7 @@ return {
                 return { "treesitter", "indent" }
             end,
             open_fold_hl_timeout = 0,
-            close_fold_kinds = { "imports", "comment" },
+            close_fold_kinds_for_ft = { "imports", "comment" },
             fold_virt_text_handler = function(virtText, lnum, endLnum, width, truncate)
                 local newVirtText = {}
                 local totalLines = vim.api.nvim_buf_line_count(0)