Added coloring by HMM, fixed discrepancy issue