Small game on Python