Choose only relevant features for model fitting