added support for filtering trees when augmenting