making nn.py make sense