Minor changes
diff --git a/src/main/java/org/apache/horn/distbelief/DistBeliefModelTrainer.java b/src/main/java/org/apache/horn/distbelief/DistBeliefModelTrainer.java
index a19139c..632e4c2 100644
--- a/src/main/java/org/apache/horn/distbelief/DistBeliefModelTrainer.java
+++ b/src/main/java/org/apache/horn/distbelief/DistBeliefModelTrainer.java
@@ -61,7 +61,7 @@
         sync();
       }
         
-      // calculate total error
+      // 2.2 calculate total error
       sync();
       
       // 2.3 downward propagation (start from the total error)
@@ -70,8 +70,8 @@
         sync();
       }
     
-      // calculate the the average training error
     }
+    // calculate the the average training error
     */
     
   }
@@ -81,7 +81,7 @@
   }
 
   private void pushParameters(BSPPeer peer) {
-    // TODO push updated weights     
+    // TODO push updated weights
   }
 
 }