diff --git a/examples/cifar_distributed_cnn/train_cnn.py b/examples/cifar_distributed_cnn/train_cnn.py
old mode 100644
new mode 100755
index 26e0403..d102623
--- a/examples/cifar_distributed_cnn/train_cnn.py
+++ b/examples/cifar_distributed_cnn/train_cnn.py
@@ -170,9 +170,6 @@
                                                    train_x, train_y, val_x,
                                                    val_y)
 
-
-
-
     if model.dimension == 4:
         tx = tensor.Tensor(
             (batch_size, num_channels, model.input_size, model.input_size), dev,
@@ -192,6 +189,7 @@
     dev.SetVerbosity(verbosity)
     model.train()
 
+    # Augmentation is done only once before training
     b = 0
     x = train_x[idx[b * batch_size:(b + 1) * batch_size]]
     if model.dimension == 4: