updated for feature selection