Improve the AST traversal and the test coverage.
Merge pull request #2 from dfellis/improve-tree-traversal-and-tests

Improve the AST traversal and the test coverage.