fix nn.py and script.py
fix nn.py and script.py