simple python game code