Add variables to return the shape and what is represented in the state. ```python >>> env.state_space() (1, 2) ```