Expanded on implementation of StateSpace