[Puzzle] Added getState and setState