updated for new feature selection code