1 files changed, 4 insertions, 0 deletions
diff --git a/lib/Lexer/Lexer.cc b/lib/Lexer/Lexer.cc
index 4f7f997..a71b37c 100644
--- a/lib/Lexer/Lexer.cc
+++ b/lib/Lexer/Lexer.cc
@@ -132,6 +132,10 @@ int Lexer::gettok() {
         if (IdentifierStr == "ENDTYPE") return tok_endtype;
         if (IdentifierStr == "SET") return tok_set;
         if (IdentifierStr == "DEFINE") return tok_define;
+        if (IdentifierStr == "CLASS") return tok_class;
+        if (IdentifierStr == "ENDCLASS") return tok_endclass;
+        if (IdentifierStr == "PUBLIC") return tok_public;
+        if (IdentifierStr == "PRIVATE") return tok_private;
 
         return tok_identifier;
     }