Update mathlib.py