Add option to dump the state space