A little game in python