diff --git a/README.md b/README.md
index e978aae..79a353e 100644
--- a/README.md
+++ b/README.md
@@ -61,9 +61,10 @@ forecast = pipeline.predict(context, prediction_length)  # shape [num_series, nu
 forecast_index = range(len(df), len(df) + prediction_length)
 low, median, high = np.quantile(forecast[0].numpy(), [0.1, 0.5, 0.9], axis=0)
 
+plt.figure(figsize=(8, 4))
 plt.plot(df["#Passengers"], color="royalblue", label="historical data")
-plt.fill_between(forecast_index, low, high, color="tomato", alpha=0.3, label="80% prediction interval")
 plt.plot(forecast_index, median, color="tomato", label="median forecast")
+plt.fill_between(forecast_index, low, high, color="tomato", alpha=0.3, label="80% prediction interval")
 plt.legend()
 plt.grid()
 plt.show()