added the test for iris data back to the neural network