diff --git a/toolkit/reportSummary/energyAnalysis.py b/toolkit/reportSummary/energyAnalysis.py
index 86d92d713f82b772488c0b7f41dd3f5c48a4f865..a877cdf66496ac8edb4af15a80af8de4878fdb48 100644
--- a/toolkit/reportSummary/energyAnalysis.py
+++ b/toolkit/reportSummary/energyAnalysis.py
@@ -60,14 +60,13 @@ class EnergyData:
         self.currentTotalHosts += 1
         if energyLevel < 0.1:
             self.currentHostsThatEverReachedLowEnergyLimit.add(host)
-            if(energyLevel == 0 ):
-                # Draws two functions over the same x values.
+            if energyLevel == 0:
                 self.currentHostsThatEverReachedZeroEnergy.add(host)
                 self.currentZeroEnergyHosts += 1
             else:
                 self.currentLowEnergyHosts += 1
 
-# Draws two functions over the same x values.
+# Draws the four functions over the same x values.
 # Labels are selected as appropriate for energy analysis.
 # In the end saves the plot
 def drawAndSafePlots(x, y_lowEnergy, y_noEnergy, y_totalLowEnergy, y_totalNoEnergy, graphicFileName):