Fun game in python