modifide lin_regression