Small game in python