misc testing improvements
improve NN visualisation