Random math fun with Python