diff --git a/README.md b/README.md
index 3c6acca..11e1f4d 100644
--- a/README.md
+++ b/README.md
@@ -17,7 +17,9 @@ Python packages:
 ## Usage
 
 ### Download weights
-    
+Download manually from [here](https://github.com/d-li14/mobilenetv3.pytorch/raw/master/pretrained/mobilenetv3-large-657e7b3d.pth)  
+or use  
+
     ./download_weight.sh
 
 ### Run the script
diff --git a/download_weight.sh b/download_weight.sh
index 876bc25..59f7ffd 100755
--- a/download_weight.sh
+++ b/download_weight.sh
@@ -1,7 +1,6 @@
 #!/usr/bin/env bash
 
-git clone git@github.com:d-li14/mobilenetv3.pytorch.git ./tmp
-
 mkdir pretrained
-cp ./tmp/pretrained/* ./pretrained/
-rm -rf ./tmp
\ No newline at end of file
+cd pretrained
+wget https://github.com/d-li14/mobilenetv3.pytorch/raw/master/pretrained/mobilenetv3-large-657e7b3d.pth
+
diff --git a/main.py b/main.py
index 217112c..94e1e0d 100644
--- a/main.py
+++ b/main.py
@@ -10,6 +10,7 @@
 
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
+logger.addHandler(logging.StreamHandler()) 
 
 
 def main():