Add python code for the game