code: def states(self):
          
          screen = self.env.getScreenRGB()
          return dict(shape=screen.shape, type='int')
docs: Return the state space. Might include subdicts if multiple states are
          available simultaneously.
  
          Returns: dict of state properties (shape and type).