Improved HMM plot. Added prediction based on states.

Current implementation quite slow, probably due to number of
states/iteration. Need to formulate data for HMM better