expectation maximization now runs but doesnt seem to work so great