Linear Fitting In Python