[ "fun" "λ" "forall" "∏" ] @keyword
(axiom) @keyword

[ "->" "=>" "→" "⇒" "," ":=" ";" ":" ] @punctuation.delimiter

[(star) (square)] @constant.builtin

(comment) @comment
(block_comment) @comment

[ "(" ")" ] @punctuation.bracket

(definition
  name: (identifier) @function)

(param_block
  param: (identifier)+ @variable.parameter)

(param_block
  type: (expr) @type)

(ascription
  type: (expr) @type)