function DAG:updateGradInput(input, gradOutput)
    self:sort()
 
-   self:applyOnModules(function(m, i, go) m:updateGradInput(i, go) end, self.outputModules, input, gradOutput)
+   self:applyOnModules(
+      function(m, i, go) m:updateGradInput(i, go) end,
+      self.outputModules, input, gradOutput
+   )
 
    for k = self.sorted, 1, -1 do
       local m = sorted[k]
 
 
 ----------------------------------------------------------------------
 
--- gradInput = g:updateGradInput({ input }, output)
+gradInput = g:updateGradInput({ input }, output)