Cleaned up doc comments
Re-added TreeBuilder, was erroneously removed