addition of feature selection function