Clean up predict.m