Updated MNIST example