Small python game