Add fit method