Update nas.py