# generate line of best fit
m, b = np.polyfit(X[:, 2], y, 1)
best_fit = m*X + b