add neural network to increase performance