diff --git a/lua/plugins/treesitter.lua b/lua/plugins/treesitter.lua
index caacf0f..41a1e0d 100644
--- a/lua/plugins/treesitter.lua
+++ b/lua/plugins/treesitter.lua
@@ -7,6 +7,7 @@ return {
     -- TODO: Fork this and add CMake support
     -- TODO: Add shell support (sh/bash/zsh if/for/while/etc)
     'RRethy/nvim-treesitter-endwise',
+    'danymat/neogen',
   },
   build = ':TSUpdate',
   config = function()
@@ -99,9 +100,8 @@ return {
       },
     })
 
-    -- Keymaps
-    vim.keymap.set('n', '<leader>fm', function()
-      vim.o.foldmethod = 'expr'
-    end, {})
+    require('neogen').setup({ snippet_engine = 'luasnip' })
+    vim.keymap.set('n', '<leader>d', require('neogen').generate,
+      { desc = 'Generate documentation' })
   end
 }