matlab code for prediction