diff --git a/oc_module/pyramid_oc_block.py b/oc_module/pyramid_oc_block.py
index 5d11cd3..421f3a8 100755
--- a/oc_module/pyramid_oc_block.py
+++ b/oc_module/pyramid_oc_block.py
@@ -72,18 +72,18 @@ def __init__(self, in_channels, key_channels, value_channels, out_channels=None,
     def forward(self, x):
         batch_size, c, h, w = x.size(0), x.size(1), x.size(2), x.size(3)
 
-        local_x = []
+        local_x = [] 
         local_y = []
-        step_h, step_w = h//self.scale, w//self.scale
-        for i in range(0, self.scale):
-            for j in range(0, self.scale):
-                start_x, start_y = i*step_h, j*step_w
-                end_x, end_y = min(start_x+step_h, h), min(start_y+step_w, w)
-                if i == (self.scale-1):
+        step_h, step_w = h//self.scale, w//self.scale 
+        for i in range(0, self.scale): 
+            for j in range(0, self.scale): 
+                start_x, start_y = i*step_h, j*step_w 
+                end_x, end_y = min(start_x+step_h, h), min(start_y+step_w, w) 
+                if i == (self.scale-1): 
                     end_x = h
                 if j == (self.scale-1):
                     end_y = w          
-                local_x += [start_x, end_x]
+                local_x += [start_x, end_x] 
                 local_y += [start_y, end_y]
 
         value = self.f_value(x)
@@ -91,11 +91,11 @@ def forward(self, x):
         key = self.f_key(x)
 
         local_list = []
-        local_block_cnt = 2*self.scale*self.scale
+        local_block_cnt = 2*self.scale*self.scale 
         for i in range(0, local_block_cnt, 2):
             value_local = value[:,:,local_x[i]:local_x[i+1],local_y[i]:local_y[i+1]]
             query_local = query[:,:,local_x[i]:local_x[i+1],local_y[i]:local_y[i+1]]
-            key_local = key[:,:,local_x[i]:local_x[i+1],local_y[i]:local_y[i+1]]
+              key_local = key[:,:,local_x[i]:local_x[i+1],local_y[i]:local_y[i+1]]
 
             h_local, w_local = value_local.size(2), value_local.size(3)
             value_local = value_local.contiguous().view(batch_size, self.value_channels, -1)