Add check zero class set size in finish tree.
Add comments to tests.
Fix style.
Minor style fix.