Update Predictors to work with Multi speed env

Update the predictor to incorporate the multi speed option of the environment