add feature selection