added the step for feature selection before training