update

Author: VectorZhao

README.md

@@ -37,24 +37,25 @@ Install the requirments for predicing in the current Conda environment:
 
     $ pip install -r requirements.txt
 
-### Step 6:
-Open `Jupyter Notebook` and load the file `MDN_two_planets_prediction.ipynb`:
-
-    $ jupyter notebook
-
-At this point you are ready to start investigating the interiors of rocky exoplanets!
-
 ## Usage
 
 ```python
-from deepexo import rockyplanet
-
-# Kepler-78b
+from deepexo.rockyplanet import RockyPlanet
+# Keplar-78b
 M = 1.77  # mass in Earth masses
 R = 1.228  # radius in Earth radii
 cFeMg = 0.685  # bulk Fe/(Mg + Si) (molar ratio)
-
-
+k2 = 0.819  # tide Love number
+
+planet_params = [
+    M,
+    R,
+    cFeMg,
+    k2,
+]
+rp = RockyPlanet()
+pred = rp.predict(planet_params)
+rp.plot(pred, save=True, filename="pred.png")
 ```
 ## References
 [https://www.science.org/doi/10.1126/science.abg8794](https://www.science.org/doi/10.1126/science.abg8794)