HMM

Problem 1 and 2 is working now