Update NEW_BASE.py