Update NeuralNetAI.java