1 files changed, 5 insertions, 0 deletions

diff --git a/12/main.l b/12/main.l
index 2fa1b44..5641fad 100644
--- a/12/main.l
+++ b/12/main.l
@@ -12,6 +12,7 @@ extern int yylval;
 "long" return LONG;
 "return" return RET;
 "for" return FOR;
+"while" return WHILE;
 "if" return IF;
 "else" return ELSE;
 "," return COMMA;
@@ -33,6 +34,10 @@ extern int yylval;
 ">" return GT;
 "<=" return LTE;
 ">=" return GTE;
+[0-9]+ {
+    yylval = atoi(yytext);
+    return NUM;
+};
 [a-zA-Z_][a-zA-Z_0-9]* return ID;
 . ;
 \n ;