Small game with pygame