updated Policies according to new StateSpace class