made minor changes to state space