1 files changed, 16 insertions, 1 deletions
diff --git a/src/ast/nodes/ForStmtNode.java b/src/ast/nodes/ForStmtNode.java
index 507d314..629d67e 100644
--- a/src/ast/nodes/ForStmtNode.java
+++ b/src/ast/nodes/ForStmtNode.java
@@ -73,8 +73,23 @@ public class ForStmtNode implements Node {
     }
 
     @Override
+    public String printAST(String prefix) {
+        return prefix + "For\n" + exprList.printAST(prefix + "  ") + block.printAST(prefix + "  ");
+    }
+
+    @Override
     public String toPrint(String prefix) {
-        return prefix + "For\n" + exprList.toPrint(prefix + "  ") + block.toPrint(prefix + "  ");
+        String str = prefix + "for ";
+        str += exprList.toPrint("") + ":\n";
+        str += block.toPrint(prefix + "\t");
+        return str;
     }
 
+    public Node getBlock() {
+        return block;
+    }
+
+    public Node getExprList() {
+        return exprList;
+    }
 }