Update quantity.py