diff --git a/.idea/misc.xml b/.idea/misc.xml
index f67039df..c974bd55 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,5 +3,5 @@
   <component name="JavaScriptSettings">
     <option name="languageLevel" value="ES6" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (torch-1.4-venv) (2)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (torch-venv)" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/.idea/mmsr.iml b/.idea/mmsr.iml
index 817b8ac1..3bd5c1db 100644
--- a/.idea/mmsr.iml
+++ b/.idea/mmsr.iml
@@ -9,7 +9,7 @@
       <excludeFolder url="file://$MODULE_DIR$/results" />
       <excludeFolder url="file://$MODULE_DIR$/tb_logger" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.7 (torch-1.4-venv)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.7 (torch-venv)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="module" module-name="SwitchedConvolutions" />
   </component>
diff --git a/codes/requirements.txt b/codes/requirements.txt
index c28aebf0..15125658 100644
--- a/codes/requirements.txt
+++ b/codes/requirements.txt
@@ -6,3 +6,5 @@ tb-nightly
 future
 scp
 tqdm
+matplotlib
+scipy