Simple python game code