Maybe we could avoid some of the problems that @lilipeng was having by renumbering the states back to `{0,1,...n_states-1}` here. http://msmbuilder.org/api/generated/msmbuilder.MSMLib.renumber_states.html