diff --git a/codes/models/archs/feature_arch.py b/codes/models/archs/feature_arch.py
index 3cafcc0d..80f77edb 100644
--- a/codes/models/archs/feature_arch.py
+++ b/codes/models/archs/feature_arch.py
@@ -61,8 +61,4 @@ class WideResnetFeatureExtractor(nn.Module):
         x = self.model.layer1(x)
         x = self.model.layer2(x)
         x = self.model.layer3(x)
-        return x
-
-
-w = WideResnetFeatureExtractor()
-w.forward(torch.randn(3,64,64))
\ No newline at end of file
+        return x
\ No newline at end of file
diff --git a/codes/process_video.py b/codes/process_video.py
index df6046e8..c580543f 100644
--- a/codes/process_video.py
+++ b/codes/process_video.py
@@ -38,8 +38,10 @@ class FfmpegBackedVideoDataset(data.Dataset):
     def get_time_for_it(self, it):
         secs = it / self.frame_rate + self.start_at
         mins = int(secs / 60)
+        hours = int(mins / 60)
         secs = secs - (mins * 60)
-        return '%02d:%06.3f' % (mins, secs)
+        mins = mins % 60
+        return '%02d:%02d:%06.3f' % (hours, mins, secs)
 
     def __getitem__(self, index):
         if self.vertical_splits > 0: