--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-TOAD 🐸 <img src="https://github.com/mahmoodlab/TOAD/blob/master/UNP.png?raw=true" width="325px" align="right" />
+TOAD 🐸 <img src="https://github.com/mahmoodlab/TOAD/blob/master/docs/UNP.png?raw=true" width="325px" align="right" />
 ===========
 ### AI-based Pathology Predicts Origins for Cancers of Unknown Primary 
 *Nature*
@@ -114,7 +114,7 @@
 
 To test trained models on your own custom datasets, you can add them into **eval_mtl_concat.py**, the same way as you do for **main_mtl_concat.py**.
 
-<img src="https://github.com/mahmoodlab/TOAD/blob/master/github_heatmap.jpg?raw=true" width="1000px" align="center" />
+<img src="https://github.com/mahmoodlab/TOAD/blob/master/docs/github_heatmap.jpg?raw=true" width="1000px" align="center" />
 
 ## Issues
 - Please report all issues on the public forum.