changes

Author: VectorZhao

README.md

@@ -41,6 +41,7 @@ Install the requirments for predicing in the current Conda environment:
 
 ```python
 from deepexo.rockyplanet import RockyPlanet
+
 # Keplar-78b
 M = 1.77  # mass in Earth masses
 R = 1.228  # radius in Earth radii
@@ -55,7 +56,12 @@ planet_params = [
 ]
 rp = RockyPlanet()
 pred = rp.predict(planet_params)
-rp.plot(pred, save=True, filename="pred.png")
+rp.plot(pred,
+        save=True,  # save to file
+        filename="pred.png"  # save to current directory, filename is pred.png, you can change the extension to eps
+        # or pdf
+        )
+
 ```
 ## References
 [https://www.science.org/doi/10.1126/science.abg8794](https://www.science.org/doi/10.1126/science.abg8794)