diff --git a/init.vim b/init.vim
index 348ad9a..a915825 100644
--- a/init.vim
+++ b/init.vim
@@ -74,6 +74,7 @@ endif
 call plug#end()
 
 colorscheme tokyonight
+syntax off
 
 let mapleader = " "
 
diff --git a/lua/init.lua b/lua/init.lua
index 28b99b4..119d99c 100644
--- a/lua/init.lua
+++ b/lua/init.lua
@@ -14,6 +14,12 @@ local ok_oil, oil = pcall(require, "oil")
 
 vim.opt.completeopt = { "menu", "menuone", "noselect" }
 
+vim.filetype.add({
+  extension = {
+    jai = "jai",
+  },
+})
+
 local function format_tex_buffer()
   local result = vim.system({ "latexindent" }, {
     stdin = table.concat(vim.api.nvim_buf_get_lines(0, 0, -1, false), "\n") .. "\n",
@@ -267,10 +273,10 @@ for server, config in pairs(servers) do
   vim.lsp.enable(server)
 end
 
-require("nvim-treesitter.configs").setup({
-  sync_install = false,
-  highlight = {
-    enable = true,
-    additional_vim_regex_highlighting = false,
-  },
+vim.api.nvim_create_autocmd("FileType", {
+  group = vim.api.nvim_create_augroup("UserTreesitterHighlight", { clear = true }),
+  callback = function(args)
+    pcall(vim.treesitter.start, args.buf)
+  end,
+  desc = "Start Tree-sitter highlighting when a parser is available",
 })