changed input arguments of predictor