Python Simple Game Code