Simple HMM package in Python