Updates to CIFAR-10 loader.