Add test for neural net grid search