diff --git a/config.py b/config.py
index 015b69d..68cd621 100755
--- a/config.py
+++ b/config.py
@@ -24,7 +24,7 @@ def get_args():
                         help='feature extractor: wide_resnet50_2')
     parser.add_argument('-wb', '--backbone_weights', default='', type=str, metavar='C', 
                         help='path to Wide resnet 50 weights if not specified \
-                        /https://download.pytorch.org/models/wide_resnet50_2-95faca4d.pth will be downloaded')
+                        https://download.pytorch.org/models/wide_resnet50_2-95faca4d.pth will be downloaded')
     parser.add_argument('-dec', '--dec-arch', default='freia-cflow', type=str, metavar='A',
                         help='normalizing flow model (default: freia-cflow)')
     parser.add_argument('-pl', '--pool-layers', default=3, type=int, metavar='L',
diff --git a/custom_models/resnet.py b/custom_models/resnet.py
index 52edfdd..f2de35f 100644
--- a/custom_models/resnet.py
+++ b/custom_models/resnet.py
@@ -17,7 +17,7 @@
     'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth',
     'resnext50_32x4d': 'https://download.pytorch.org/models/resnext50_32x4d-7cdf4587.pth',
     'resnext101_32x8d': 'https://download.pytorch.org/models/resnext101_32x8d-8ba56ff5.pth',
-    'wide_resnet50_2': '/https://download.pytorch.org/models/wide_resnet50_2-95faca4d.pth',
+    'wide_resnet50_2': 'https://download.pytorch.org/models/wide_resnet50_2-95faca4d.pth',
     'wide_resnet101_2': 'https://download.pytorch.org/models/wide_resnet101_2-32ee1156.pth',
 }
 
diff --git a/main.py b/main.py
index de0a2b5..6f712c9 100755
--- a/main.py
+++ b/main.py
@@ -2,6 +2,7 @@
 import os, time, random, math
 import numpy as np
 import torch
+import datetime
 import torch.nn.functional as F
 from sklearn.metrics import roc_auc_score, auc, precision_recall_curve
 from skimage.measure import label, regionprops