Update nn.py

added some runner code