Add use of neural network and training