diff --git a/README.md b/README.md
index dc0af55..491d4ac 100644
--- a/README.md
+++ b/README.md
@@ -90,7 +90,7 @@ criterion:
   type: <loss_function>
 ```
 
-**WARNING:** Processing the **PEMSD7** dataset may require more than **32GB** of memory. If your system lacks sufficient memory, it is recommended to increase the size of the swap partition.
+**WARNING:** Processing the **PEMSD7** dataset may require more than **32GB** RAM. If your system lacks sufficient RAM, it is recommended to increase the size of the swap partition.
 
 
 
@@ -127,6 +127,12 @@ python exp/global.py
 
 The script will generate a **baseline.jpg** file to visualize the logs. You are also free to modify the script to implement additional functionality as needed.
 
+You may install matplotlib first for drawing:
+
+```
+pip install matplotlib
+```
+
 
 
 # Citation