1 files changed, 2 insertions, 0 deletions
diff --git a/include/cps/CodeGen.h b/include/cps/CodeGen.h
index 57f0899..832ba7a 100644
--- a/include/cps/CodeGen.h
+++ b/include/cps/CodeGen.h
@@ -4,6 +4,7 @@
 #include "llvm/IR/Module.h"
 #include "cps/AST.h"
 #include "cps/RuntimeCheck.h"
+#include "cps/FunctionGen.h"
 #include <map>
 #include <memory>
 
@@ -19,6 +20,7 @@ class CodeGen {
     
     std::unique_ptr<ArrayHandler> Arrays;
     std::unique_ptr<RuntimeCheck> RuntimeChecker;
+    std::unique_ptr<FunctionGen> FuncGen;
 
     llvm::FunctionCallee PrintfFunc;
     llvm::FunctionCallee ScanfFunc;