From 477d826a9a2ed7ada7d334a87f9bead497c83401 Mon Sep 17 00:00:00 2001
From: JSChalmers <44227163+JSChalmers@users.noreply.github.com>
Date: Mon, 8 Mar 2021 19:24:04 +0100
Subject: [PATCH] Update README.md

---
 README.md | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 3b35852..5d89541 100644
--- a/README.md
+++ b/README.md
@@ -1 +1,7 @@
-# deep-learning-based-MIMO
\ No newline at end of file
+# End-to-end Learning of MIMO and Multi-user Communication
+
+
+This repository is based on joint work with Christian Häger, Jochen Schrodör, Timothy J. O'Shea, Erik Agrell and Henk Wymeersch. If you decide to use the source code for your research, please make sure to cite our paper(s):
+
+J. Song, C. Häger, J. Schröder, T. O’Shea, and H. Wymeersch, “Bench-marking end-to-end learning of MIMO physical-layer communication,”inProc.  IEEE  Glob.  Communication  Conf.  (GLOBECOM),  Taipei,Taiwan, 2020
+