Update nn.py

In line 51, would "(Theta ** 2).sum()" work better?