Linear regression with SciPy and NumPy