Fixed another issue with mnist loader.

diff --git a/code/mnist_loader.py b/code/mnist_loader.py
index feeb8c7..ed1f1d8 100644
--- a/code/mnist_loader.py
+++ b/code/mnist_loader.py
@@ -173,7 +173,6 @@ def load_mnist_2D(filename = ''):
     test_inputs = np.array([x.reshape(28, 28, 1) for x in te_d])

     # Return the data.
-    return training_inputs, ku.to_categorical(tr_v, 10), \
-        validation_inputs, ku.to_categorical(va_v, 10), \
-        test_inputs, ku.to_categorical(te_v, 10)
+    return training_inputs, tr_v, validation_inputs, va_v, \
+        test_inputs, te_v