completed neural network task