Update LinearRegressionDriver.java