R code for neural network on Iris data