Working on two urns example, up to writing E step