4 files changed, 16 insertions, 0 deletions
diff --git a/main.c b/main.c
index 34acd52..15135bb 100644
--- a/main.c
+++ b/main.c
@@ -193,6 +193,7 @@ int main()
 	assert(cur_scope);
 
 	yyparse();
+	free_scope(cur_scope);
 
 	return 0;
 }
diff --git a/pc.y b/pc.y
index d679c18..426cf71 100644
--- a/pc.y
+++ b/pc.y
@@ -97,6 +97,7 @@ program
 	{
 		set_ret_type($9);
 		print_tree($9);
+		free_tree($4);
 	}
 ;
 
@@ -164,6 +165,7 @@ sub_prog_declaration
 	{
 		set_ret_type($4);
 		print_tree($4);
+		free_tree($4);
 		pop_scope(&cur_scope);
 	}
 ;
diff --git a/tree.c b/tree.c
index fd40781..bc4a770 100644
--- a/tree.c
+++ b/tree.c
@@ -191,3 +191,14 @@ int spaces;
 	}
 
 }
+
+void free_tree(t)
+ptree *t;
+{
+	if (!t)
+		return;
+
+	free_tree(t->l);
+	free_tree(t->r);
+	free(t);
+}
diff --git a/tree.h b/tree.h
index 23edf8f..ed7c1ef 100644
--- a/tree.h
+++ b/tree.h
@@ -30,4 +30,6 @@ ptree* mkop(int, int, ptree*, ptree*);
 void update_type_info(ptree*, int);
 void set_ret_type(ptree*);
 
+void free_tree(ptree*);
+
 #endif