Update formula.py