Add a method for getting the known states from a model