Add selection based on mgain
Select nodes based on their score, not marginal gain