A3 - need to complete predict.m