clean up some of the neural network tests