Linear Regression Graph Code In Python