Added code for predictions.