1 files changed, 38 insertions, 10 deletions
diff --git a/src/ast/nodes/SimpleStmtNode.java b/src/ast/nodes/SimpleStmtNode.java
index 31a935f..3952242 100644
--- a/src/ast/nodes/SimpleStmtNode.java
+++ b/src/ast/nodes/SimpleStmtNode.java
@@ -73,28 +73,56 @@ public class SimpleStmtNode implements Node {
     }
 
     @Override
-    public String toPrint(String prefix) {
+    public String printAST(String prefix) {
         String str = prefix + "SimpleStmt\n";
 
         prefix += "  ";
 
         if (assignment != null) {
-            str += assignment.toPrint(prefix);
+            str += assignment.printAST(prefix);
+        } else if (expr != null) {
+            str += expr.printAST(prefix);
+        } else if (returnStmt != null) {
+            str += returnStmt.printAST(prefix);
+        } else if (importStmt != null) {
+            str += importStmt.printAST(prefix);
         }
 
-        if (expr != null) {
-            str += expr.toPrint(prefix);
-        }
+        return str;
+    }
 
-        if (returnStmt != null) {
-            str += returnStmt.toPrint(prefix);
-        }
+    @Override
+    public String toPrint(String prefix) {
+        
+        String str = prefix;
 
-        if (importStmt != null) {
-            str += importStmt.toPrint(prefix);
+        if (assignment != null) {
+            str += assignment.toPrint("");
+        } else if (expr != null) {
+            str += expr.toPrint("");
+        } else if (returnStmt != null) {
+            str += returnStmt.toPrint("");
+        } else if (importStmt != null) {
+            str += importStmt.toPrint("");
         }
 
         return str;
     }
 
+    public Node getImportStmt() {
+        return importStmt;
+    }
+
+    public Node getReturnStmt() {
+        return returnStmt;
+    }
+    
+    public Node getAssignment() {
+        return assignment;
+    }
+
+    public Node getExpr() {
+        return expr;
+    }
+
 }