Returning self from fit()