Finishes annotating the neural network (nn.py) code and ends chapter 4