diff --git a/LICENSE b/LICENSE.md
similarity index 100%
rename from LICENSE
rename to LICENSE.md
diff --git a/README.md b/README.md
index cfab347..60fcdd5 100644
--- a/README.md
+++ b/README.md
@@ -1,12 +1,24 @@
-# DRU-Net
+<div align="center">
+<h1 align="center">🫁 DRU-Net </h1>
+<h3 align="center">Segmentation of Non-Small Cell Lung Carcinomas
+</h3>
 
-## Introduction
+[![license](https://img.shields.io/badge/MIT-License-008000)](https://github.com/AICAN-Research/DRU-Net/blob/main/LICENSE.md)
+<a href="https://github.com/AICAN-Research/DRU-Net/blob/main/notebooks/TrainingOnWSIPatches.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+[![arXiv preprint](https://img.shields.io/badge/arXiv-preprint-D12424)](
+https://doi.org/10.48550/arXiv.2406.14287)
 
-This project presents DRU-Net: Lung carcinoma segmentation using multi-lens distortion and fusion refinement network.
+**DRU-Net** was developed the AICAN research group.
 
-## Getting started
+</div>
 
-### Setup
+# Introduction
+
+This repository contains the source code related to the manuscript _"DRU-Net: Lung carcinoma segmentation using multi-lens distortion and fusion refinement network"_ which is openly available on [arXiv](https://arxiv.org/abs/2406.14287).
+
+# Getting started
+
+## Setup
 
 1. Setup virtual environment and activate it:
 
@@ -21,7 +33,13 @@ source venv/bin/activate
 pip install -r requirements.txt
 ```
 
-### Linting
+
+<details>
+<summary>
+
+# Development</summary>
+
+## Linting
 
 First install linting dependencies:
 
@@ -40,3 +58,20 @@ Perform automatic linting by:
 ```
 sh shell/format.sh
 ```
+
+</details>
+
+## Citation
+
+If you found the source code or manuscript relevant in your research, please cite the following reference:
+
+```
+@misc{oskouei2024segmentation,
+    title={{Segmentation of Non-Small Cell Lung Carcinomas: Introducing DRU-Net and Multi-Lens Distortion}}, 
+    author={Soroush Oskouei and Marit Valla and André Pedersen and Erik Smistad and Vibeke Grotnes Dale and Maren Høibø and Sissel Gyrid Freim Wahl and Mats Dehli Haugum and Thomas Langø and Maria Paula Ramnefjell and Lars Andreas Akslen and Gabriel Kiss and Hanne Sorger},
+    year={2024},
+    eprint={2406.14287},
+    archivePrefix={arXiv},
+    doi={10.48550/arXiv.2406.14287}
+}
+```
\ No newline at end of file