Update math_mod.py