[msm/estimation] nstates feature for count_matrix function.

Fixes #24 

Default behavior of count_matrix was changed, see discussion.