Minor improvements to mnist_average_darkness