Add polynomial fitting with np.polyfit)