Switch from CIFAR-100 to CIFAR-10 for simplicity