Added TreeWalker tests