add a line in NEW.py