adding game state to python version as well