Small game (Python)