Minor change in feature selection