[FEATURE] Implement traverse method for tree
[DOC] Update docs for 0.5.2