Update NN.py