Added MLE functionality