Update HMM.py