Update model_NTD.py