diff --git a/schala-lang/language/src/ast/mod.rs b/schala-lang/language/src/ast/mod.rs
index 8e22b2a..55dda76 100644
--- a/schala-lang/language/src/ast/mod.rs
+++ b/schala-lang/language/src/ast/mod.rs
@@ -5,7 +5,7 @@ mod visitor;
 mod operators;
 
 pub use operators::{PrefixOp, BinOp};
-pub use visitor::{walk_ast, ASTVisitor};
+pub use visitor::{walk_ast, walk_block, ASTVisitor};
 
 use std::rc::Rc;
 use crate::derivative::Derivative;
@@ -119,21 +119,6 @@ pub enum Declaration {
   }
 }
 
-/*
- * @foo(arg1, arg2)
- * fn hello() {
- *
- * }
- *
- * Declaration::Annotation {
- *  name: "foo",
- *  arguments: vec!["arg1", "arg2"]
- *  }
- *
- *
- *
- */
-
 #[derive(Debug, PartialEq, Clone)]
 pub struct Signature {
   pub name: Rc<String>,
diff --git a/schala-lang/language/src/ast/visitor.rs b/schala-lang/language/src/ast/visitor.rs
index 8d16ed2..2948c82 100644
--- a/schala-lang/language/src/ast/visitor.rs
+++ b/schala-lang/language/src/ast/visitor.rs
@@ -19,7 +19,7 @@ pub fn walk_ast<V: ASTVisitor>(v: &mut V, ast: &AST) {
     walk_block(v, &ast.statements);
 }
 
-fn walk_block<V: ASTVisitor>(v: &mut V, block: &Block) {
+pub fn walk_block<V: ASTVisitor>(v: &mut V, block: &Block) {
     use StatementKind::*;
     for statement in block.iter() {
         match statement.kind {