diff --git a/lenet5.py b/lenet5.py
index 7ed8728..6ecd241 100644
--- a/lenet5.py
+++ b/lenet5.py
@@ -164,6 +164,10 @@ def main():
     test['features'], test['labels'] = read_mnist('t10k-images-idx3-ubyte.gz',
                                                   't10k-labels-idx1-ubyte.gz')
 
+    print(type(train))
+    print(type(train['features']))
+    print(type(train['labels']))
+
     # Step 3:
     # Explore the dataset
     print('Number of training images:', train['features'].shape[0])