diff --git a/README.md b/README.md
index e3e02dd..32b7e4a 100644
--- a/README.md
+++ b/README.md
@@ -9,6 +9,14 @@
 
 ---
 
+How does it work?
+----
+
+![alt text](static/architecture.png)
+
+MarioGPT is a finetuned GPT2 model (specifically, [distilgpt2](https://huggingface.co/distilgpt2)), that is trained on a subset Super Mario Bros and Super Mario Bros: The Lost Levels levels, provided by [The Video Game Level Corpus](https://github.com/TheVGLC/TheVGLC). MarioGPT is able to generate levels, guided by a simple text prompt. This generation is not perfect, but we believe this is a great first step more controllable and diverse level / environment generation.
+
+
 Requirements
 ----
 - python3.8+
diff --git a/static/architecture.png b/static/architecture.png
new file mode 100644
index 0000000..f32e85b
Binary files /dev/null and b/static/architecture.png differ