hmm.. change the way to pick feature