add nn.py