- made the neural network test a bit more configurable