Python Least Square