Add playState and tests