Added feature selection choice