Forgot to return self in fit()