Fix pseudocode.

- Make `arb` a keyword.
- Correct argument order to `map`.

1 files changed, 11 insertions, 1 deletions

diff --git a/lexer.py b/lexer.py
index 6963b77..cdce839 100644
--- a/lexer.py
+++ b/lexer.py
@@ -9,7 +9,17 @@ class SystemTLexer(OcamlLexer):
     aliases = ['syst']
     filenames = ['*.syst']
 
-    EXTRA_KEYWORDS = ['primrec', 'fold', 'foldmatch', 'roll', 'roll2']
+    EXTRA_KEYWORDS = [
+        'arb',
+        'arb)',
+        'fold',
+        'foldmatch',
+        'map',
+        'max',
+        'primrec',
+        'roll',
+        'roll2',
+    ]
 
     def get_tokens_unprocessed(self, text):
         for index, token, value in OcamlLexer.get_tokens_unprocessed(self, text):