More work on MNIST8M example