modify equip.py