Update box.py