diff --git a/ftdetect/perga.vim b/ftdetect/perga.vim
deleted file mode 100644
index 8e7e342..0000000
--- a/ftdetect/perga.vim
+++ /dev/null
@@ -1 +0,0 @@
-autocmd VimEnter,BufWinEnter,BufRead,BufNewFile *.pg setlocal filetype=perga
diff --git a/lua/perga/init.lua b/lua/perga/init.lua
index 388c40a..161f76d 100644
--- a/lua/perga/init.lua
+++ b/lua/perga/init.lua
@@ -1,6 +1,10 @@
 local M = {}
 
 function M.setup(opts)
+    vim.filetype.add({
+        extension = { pg = 'perga' }
+    })
+
     local parser_config = require "nvim-treesitter.parsers".get_parser_configs()
     parser_config.perga = {
         install_info = {
diff --git a/queries/perga/highlights.scm b/queries/perga/highlights.scm
index a44408c..c306028 100644
--- a/queries/perga/highlights.scm
+++ b/queries/perga/highlights.scm
@@ -1,8 +1,17 @@
 [ "fun" "λ" "forall" "∏" ] @keyword
 [ "->" "=>" "→" "⇒" "," ":=" ";" ":" ] @punctuation.delimiter
-[ "(" ")" ] @punctuation.bracket
-;[ "*" "□" "[]" ] @constant.builtin
+
+[(star) (square)] @constant.builtin
+
 (comment) @comment
-(identifier) @variable
-;(param) @variable.parameter
-;(type) @type
+
+[ "(" ")" ] @punctuation.bracket
+
+(definition
+  name: (identifier) @function)
+
+(param_block
+  param: (identifier)+ @variable.parameter)
+
+(param_block
+  type: (expr) @type)
diff --git a/queries/rainbow-delimiters.scm b/queries/rainbow-delimiters.scm
new file mode 100644
index 0000000..95f682d
--- /dev/null
+++ b/queries/rainbow-delimiters.scm
@@ -0,0 +1 @@
+["(" ")"] @delimiter