WIP on real MNIST test set