Changes made for predict.m