work in progress -- toy model expectation maximization