Update to test data and linear model prediction code.