Change minor code in read_cifar10.py