Create mental-math.py