diff --git a/CITATION.cff b/CITATION.cff
new file mode 100644
index 0000000..268b544
--- /dev/null
+++ b/CITATION.cff
@@ -0,0 +1,19 @@
+cff-version: 1.2.0
+title: Solution for Sensorium 2023 Competition
+message: 'If you use this work, please cite it using these metadata.'
+type: software
+authors:
+  - given-names: Ruslan
+    family-names: Baikulov
+    email: ruslan1123@gmail.com
+    orcid: 'https://orcid.org/0009-0003-4400-0619'
+repository-code: 'https://github.com/lRomul/sensorium'
+abstract: >-
+  This repository contains the winning solution for the Sensorium 2023 Competition.
+  The competition aimed to predict the activity of neurons in the primary visual
+  cortex of mice in response to videos. The proposed solution includes a novel
+  model architecture called DwiseNeuro and a training pipeline with a solid
+  cross-validation strategy and knowledge distillation.
+license: MIT
+version: v23.11.18
+date-released: '2023-11-18'
diff --git a/README.md b/README.md
index 67aa703..9fe6507 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,10 @@
+<div align="center">
+
 # Solution for Sensorium 2023 Competition
 
-![header](data/readme_images/header.png)
+[![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
+
+</div>
 
 This repository contains the winning solution for the Sensorium 2023, part of the NeurIPS 2023 competition track.
 The competition aimed to find the best model that can predict the activity of neurons in the primary visual cortex of mice in response to videos.
diff --git a/data/readme_images/header.png b/data/readme_images/header.png
deleted file mode 100644
index c0129a7..0000000
Binary files a/data/readme_images/header.png and /dev/null differ