Linear Fit In Python