Added code comments to TreeGenerator