finished EM expectation steps