1 files changed, 5 insertions, 1 deletions
diff --git a/include/cps/CodeGen.h b/include/cps/CodeGen.h
index 88e4755..9be6cc7 100644
--- a/include/cps/CodeGen.h
+++ b/include/cps/CodeGen.h
@@ -20,10 +20,14 @@ class CodeGen {
     llvm::FunctionCallee ScanfFunc;
     llvm::Value *PrintfFormatStr;
     llvm::Value *ScanfFormatStr;
+    
     llvm::Value *emitExpr(ExprAST *Expr);
     void emitStmt(StmtAST *Stmt);
-    void emitIfStmt(IfStmtAST *Stmt);
     
+    void emitIfStmt(IfStmtAST *Stmt);
+    void emitWhileStmt(WhileStmtAST *Stmt);
+    void emitRepeatStmt(RepeatStmtAST *Stmt);
+    void emitForStmt(ForStmtAST *Stmt);
 
     void SetupExternalFunctions();