The famous number game in python