Update README.md

README.md

@@ -3,15 +3,16 @@
 This is python code of generating MP2, MP3, and MP4 correlation energy density based on **one boby decomposition scheme** (https://doi.org/10.1002/qua.22017)
 
 Requirement of python packages:
-```shell
-numpy, scipy, h5py, torch, quadpy, itertools, mmap, tqdm
+```python
+numpy, scipy, h5py, (py)torch, quadpy, itertools, mmap, tqdm
 ```
 Optional of python packages:
-```shell
-pycuda, skcuda # if you use GPU for preparing Ajb
+```python
+pycuda, skcuda # if you use GPU for calculating Ajb
+(py)torch-gpu # if you use GPU for calculating t(n)
 ```
 Requirement of external packages to run example files:
-```shell
+```python
 psi4 # to generate wavefunction and FCIDUMP file
 molden2aim # to convert molden format to .WFN
 ```