Added hacky loss augmenting to EM.