Neural network added