Understanding the MNIST dataset