diff --git a/RBM/cifar10.m b/RBM/cifar10.m
index 2fc6843..fa8c5f2 100644
--- a/RBM/cifar10.m
+++ b/RBM/cifar10.m
@@ -5,8 +5,8 @@ data = d.data;
 im = double(data(:,1:1024))/256 + double(data(:,1025:2048))/256 + double(data(:,2049:3072))/256;
 im = im /3;
 
-%im2 = reshape(im(2,:), 32,32)
-%imshow(im2)
+im2 = reshape(im(2,:), 32,32)
+imshow(im2)
 
 [numTrain, numPixel] = size(im)
 
@@ -16,7 +16,7 @@ fprintf(fid, '%d\n', numPixel);
 
 for m=1:numTrain,
 	for n=1:numPixel,
-		fprintf(fid, '%f\n', im(n));
+		fprintf(fid, '%f\n', im(m, n));
 	end
 end