diff --git a/read_cifar.py b/read_cifar.py
index 9684ce66df16c84f73a361bba8c14cd97bfa7b0c..abba3fae17d1007e47e2551059dd4adf24b8e528 100644
--- a/read_cifar.py
+++ b/read_cifar.py
@@ -1,14 +1,14 @@
-import numpy
+import numpy as np
 import pickle
-def unpickle(file):
+def read_cifar_batch(file):
     with open(file, 'rb') as fo:
         dict = pickle.load(fo, encoding='bytes')
-    return dict
+        data = np.array(dict[b'data']).astype('float32')
+        labels = np.array(dict[b'labels']).astype('int64')
+    return data, labels
 
-dict1= unpickle("data/cifar-10-batches-py/data_batch_1")
+vect1= read_cifar_batch("data/cifar-10-batches-py/data_batch_1")
 
-data = [dict1[key] for key in dict1]
+#print(vect1)
 
-print(len(data[1]))
-
-#def read_cifar_batch(path) :
+#def read_cifar
\ No newline at end of file