add fit method