Update to include the selection of the final set of fetures from the full dataset