Add compiler improvement

Project-02-03-04-05/main.py

@@ -17,6 +17,7 @@ from cfa.to_dot import analysis_to_dot
 from cfg.CFG import CFG
 from vistram.tram import *
 from vistram.vistram import MachineUI
+from optimizations.optimize import optimize
 
 matplotlib.use("TkAgg")
 
@@ -129,6 +130,9 @@ if __name__ == "__main__":
         ast = yacc.parser.parse(source)
 
         if mode == 0:
+            if cnsl.prompt_confirmation("\nOptimize AST?", default="y"):
+                ast = optimize(ast)
+
             # Pretty print
             if cnsl.prompt_confirmation("\nPretty-print AST?"):
                 print("")
@@ -156,6 +160,9 @@ if __name__ == "__main__":
                 print("Rendered AST diagram.")
 
         elif mode == 1:
+            if cnsl.prompt_confirmation("\nOptimize AST?", default="y"):
+                ast = optimize(ast)
+
             tram_code = assemble(ast)
 
             # Print TRAM code