diff --git a/README.md b/README.md
index 8be3287..aa53a81 100644
--- a/README.md
+++ b/README.md
@@ -6,3 +6,7 @@ Currently investigated outcomes:
 - Functional outcome: '3M mRS 0-2'
 
 Model weights: https://doi.org/10.5281/zenodo.8195710
+
+### Reference
+If you use this work for your research, please cite this paper: https://www.nature.com/articles/s43856-024-00666-w#article-info 
+> Klug, J., Leclerc, G., Dirren, E. et al. Machine learning for early dynamic prediction of functional outcome after stroke. Commun Med 4, 232 (2024). https://doi.org/10.1038/s43856-024-00666-w