Added learning curve plotting for neural network