diff --git a/src/Main.java b/src/Main.java
index cf8e67e..5970b3e 100644
--- a/src/Main.java
+++ b/src/Main.java
@@ -102,6 +102,8 @@ public static void main(String[] args) {
                     frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
                     frame.setSize(800, 600);
                     frame.setVisible(true);
+
+                    System.out.println(ast.printAST(""));
                 }
 
                 if (execute) {
diff --git a/src/ast/nodes/WhileStmtNode.java b/src/ast/nodes/WhileStmtNode.java
index 3e8a36e..394bdd3 100644
--- a/src/ast/nodes/WhileStmtNode.java
+++ b/src/ast/nodes/WhileStmtNode.java
@@ -41,7 +41,7 @@ public String codeGeneration() {
 
     @Override
     public String printAST(String prefix) {
-        return prefix + "While\n" + expr.toPrint(prefix + "  ") + block.toPrint(prefix + "  ");
+        return prefix + "While\n" + expr.printAST(prefix + "  ") + block.printAST(prefix + "  ");
     }
 
     @Override