Minor commit, just added some comments to the neural network class.