diff --git a/lua/fugitive-ts/health.lua b/lua/fugitive-ts/health.lua
index bdeebdd..48f946b 100644
--- a/lua/fugitive-ts/health.lua
+++ b/lua/fugitive-ts/health.lua
@@ -16,8 +16,11 @@ function M.check()
     vim.health.warn('vim-fugitive not detected (required for this plugin to be useful)')
   end
 
+  ---@type string[]
   local common_langs = { 'lua', 'python', 'javascript', 'typescript', 'rust', 'go', 'c', 'cpp' }
+  ---@type string[]
   local available = {}
+  ---@type string[]
   local missing = {}
 
   for _, lang in ipairs(common_langs) do
diff --git a/lua/fugitive-ts/highlight.lua b/lua/fugitive-ts/highlight.lua
index 650436d..3c29658 100644
--- a/lua/fugitive-ts/highlight.lua
+++ b/lua/fugitive-ts/highlight.lua
@@ -1,11 +1,15 @@
 local M = {}
 
+---@param bufnr integer
+---@param ns integer
+---@param hunk fugitive-ts.Hunk
 function M.highlight_hunk(bufnr, ns, hunk)
   local lang = hunk.lang
   if not lang then
     return
   end
 
+  ---@type string[]
   local code_lines = {}
   for _, line in ipairs(hunk.lines) do
     table.insert(code_lines, line:sub(2))