diff --git a/.config/nvim/lua/plugins/neogen.lua b/.config/nvim/lua/plugins/neogen.lua
new file mode 100644
index 0000000..c031c11
--- /dev/null
+++ b/.config/nvim/lua/plugins/neogen.lua
@@ -0,0 +1,7 @@
+return {
+  -- generate annotations automatically
+  "danymat/neogen",
+  opts = {
+    snippet_engine = "luasnip",
+  },
+}
diff --git a/.config/nvim/lua/plugins/nvim-treesitter.lua b/.config/nvim/lua/plugins/nvim-treesitter.lua
index 96ea786..8ebd0e5 100644
--- a/.config/nvim/lua/plugins/nvim-treesitter.lua
+++ b/.config/nvim/lua/plugins/nvim-treesitter.lua
@@ -7,27 +7,28 @@ return {
     "windwp/nvim-ts-autotag",
     -- color brackets
     -- 'p00f/nvim-ts-rainbow',
+    "nvim-treesitter/nvim-treesitter-context",
   },
   config = function()
     local treesitter = require("nvim-treesitter")
     treesitter.setup()
     treesitter.install({
-        "bash",
-        "css",
-        "html",
-        "markdown",
-        "markdown_inline",
-        "latex",
-        "python",
-        "rust",
-        "lua",
-        "yaml",
+      "bash",
+      "css",
+      "html",
+      "markdown",
+      "markdown_inline",
+      "latex",
+      "python",
+      "rust",
+      "lua",
+      "yaml",
     })
     require("nvim-ts-autotag").setup({
       opts = {
         enable_close = true,
         enable_rename = true,
-      }
+      },
     })
     -- WARN: not directly compatible anymore
     -- highlight = { enable = true },