Python Quadratic Fit