Python based CIFAR-dataset classification