diff --git a/README.md b/README.md
index 84c1664c21164a8130e77990d5f947bdbb5cdda7..bf4630fd6685f36b56e43e07bc5856af5d13dff5 100644
--- a/README.md
+++ b/README.md
@@ -21,7 +21,13 @@ cd Video_Prediction_SAVP
 pip install -r requirements.txt
 ```
 
-###Set-up on JUWELS
+### Miscellaneous installation considerations
+- In python >= 3.6, make sure to add the root directory to the PYTHONPATH`, e.g. `export PYTHONPATH=path/to/video_prediction_savp`.
+- For the best speed and experimental results, we recommend using cudnn version 7.3.0.29 and any tensorflow version >= 1.9 and <= 1.12. The final training loss is worse when using cudnn versions 7.3.1.20 or 7.4.1.5, compared to when using versions 7.3.0.29 and below.
+- Add the directories lpips-tensorflow and hickle (get from [Workflow project](https://gitlab.version.fz-juelich.de/gong1/workflow_parallel_frame_prediction) to the  `PATHONPATH `, e.g export PYTHONPATH=path/to/lpips-tensorflow
+
+
+### Set-up on JUWELS
 
 - Set up env and install packages
 
@@ -30,13 +36,6 @@ cd env_setup
 ./create_env.sh <USER_FOLDER>
 ```
 
-
-
-### Miscellaneous installation considerations
-- In python >= 3.6, make sure to add the root directory to the PYTHONPATH`, e.g. `export PYTHONPATH=path/to/video_prediction_savp`.
-- For the best speed and experimental results, we recommend using cudnn version 7.3.0.29 and any tensorflow version >= 1.9 and <= 1.12. The final training loss is worse when using cudnn versions 7.3.1.20 or 7.4.1.5, compared to when using versions 7.3.0.29 and below.
-- Add the directories lpips-tensorflow and hickle (get from [Workflow project](https://gitlab.version.fz-juelich.de/gong1/workflow_parallel_frame_prediction) to the  `PATHONPATH `, e.g export PYTHONPATH=path/to/lpips-tensorflow
-- You may need install packages by pip on JUWELS/JURECA, followed the installation instruction from [Workflow project](https://gitlab.version.fz-juelich.de/gong1/workflow_parallel_frame_prediction) 
 ### Download data
 
 - Download the ERA5 data (.hkl) from the output of DataPreprocess in the [Workflow project](https://gitlab.version.fz-juelich.de/gong1/workflow_parallel_frame_prediction)
diff --git a/env_setup/create_env.sh b/env_setup/create_env.sh
index 85ef0ab7edef985f5de5ed40ee1fa9e4ad61d5a8..6e537909497664612adbf23166fdc91770efa211 100755
--- a/env_setup/create_env.sh
+++ b/env_setup/create_env.sh
@@ -21,5 +21,6 @@ cp  -r /p/project/deepacf/deeprain/bing/hickle .
 export PYTHONPATH=${ENV_DIR}/hickle/lib/python3.6/site-packages:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}:$PYTHONPATH
 export PYTHONPATH=/p/home/jusers/${USER}/juwels/.local/bin:$PYTHONPATH
+export PYTHONPATH=${WORKING_DIR}/lpips-tensorflow:$PYTHONPATH