Test on MNIST data set