Ordinary Least Squares in Python (Challenge)