only change on model.fit method