diff --git a/docs/lstm_with_context.png b/docs/lstm_with_context.png
index 0546b12..3441299 100644
Binary files a/docs/lstm_with_context.png and b/docs/lstm_with_context.png differ
diff --git a/readme.md b/readme.md
index 6ee16dc..899af0d 100644
--- a/readme.md
+++ b/readme.md
@@ -41,8 +41,12 @@ I chose a a difficult example below, it's a window in the test set that deviates
 
 Compare this to a quick LSTM baseline below, which didn't predict this divergance from the pattern. (Bear in mind that I didn't tweak this model as much). The uncertainty and prediction are also less smooth and the log probability is lower.
 
+An LSTM with an encoder style similar to ANP's:
+
 ![](docs/lstm_with_context.png)
 
+and a normal LSTM:
+
 ![](docs/lstm_baseline.png)
 
 ## Code