Modify to predicting time