diff --git a/README.md b/README.md
index f55162c..ab2c2bb 100644
--- a/README.md
+++ b/README.md
@@ -6,7 +6,7 @@ model and motivation, please refer to our official [paper](TODO).
 
 # Model
 
-![CEM Architecture](figures/cem.png)
+![CEM Architecture](figures/cem_white_background.png)
 
 [Concept Bottleneck Models (CBMs)](https://arxiv.org/abs/2007.04612) have recently gained attention as
 high-performing and interpretable neural architectures that can explain their
diff --git a/figures/cem_white_background.png b/figures/cem_white_background.png
new file mode 100644
index 0000000..1b13cbd
Binary files /dev/null and b/figures/cem_white_background.png differ