Add method to automatically set all of the required traces for prediction.