Added fit_mle function