Adding in testNN function to do prediction on test data