Added MNIST loading.