Complete the nn.py