EM model working, TODO check different version/ add various random start