Linear Regression Visualization In Python