Rewrote Linear Regression and Logistic