diff --git a/nvim/lua/plugins/util/neogen.lua b/nvim/lua/plugins/util/neogen.lua
index 1737bf9..ad98297 100755
--- a/nvim/lua/plugins/util/neogen.lua
+++ b/nvim/lua/plugins/util/neogen.lua
@@ -2,7 +2,16 @@ return {
     "danymat/neogen",
     config = function()
         local opts = { noremap = true, silent = true, desc = "Generate docs" }
-        vim.api.nvim_set_keymap("n", "<leader>gd", ":lua require('neogen').generate()<CR>", opts)
+        vim.keymap.set("n", "<leader>gd", ":lua require('neogen').generate()<CR>", opts)
+
+        opts.desc = "Generate function docs"
+        vim.keymap.set("n", "<leader>gg", ":lua require('neogen').generate( { type = 'func' } )<CR>", opts)
+
+        opts.desc = "Generate type docs"
+        vim.keymap.set("n", "<leader>gt", ":lua require('neogen').generate( { type = 'type' } )<CR>", opts)
+
+        opts.desc = "Generate class docs"
+        vim.keymap.set("n", "<leader>gc", ":lua require('neogen').generate( { type = 'class' } )<CR>", opts)
         require("neogen").setup({
             snippet_engine = "luasnip",
         })