minor changes to feature selection