Added method visualize_hidden_state_probabilities