Update simple_math.py