* The neural network actually gets trained now