Update EM.py, add E-step & M-step