ler 0.3.7__py3-none-any.whl → 0.3.9__py3-none-any.whl

ler/rates/ler.py

@@ -15,7 +15,7 @@ from ..utils import load_json, append_json, get_param_from_json, batch_handler,
 
 
 class LeR(LensGalaxyParameterDistribution):
-    """Class to calculate both the rates of lensed and unlensed events. Please note that parameters of the simulated events are stored in json file but not as an attribute of the class. This saves RAM memory. 
+    """Class to sample of lensed and unlensed events and calculate it's rates. Please note that parameters of the simulated events are stored in json file but not as an attribute of the class. This saves RAM memory. 
 
     Parameters
     ----------
@@ -29,25 +29,54 @@ class LeR(LensGalaxyParameterDistribution):
     z_max : `float`
         maximum redshift.
         default z_max = 10.
-        for popI_II, popIII, primordial, BNS z_max = 10., 40., 40., 2. respectively.
+        for popI_II, popIII, primordial, BNS z_max = 10., 40., 40., 5. respectively.
+    event_type : `str`
+        type of event to generate.
+        default event_type = 'BBH'. Other options are 'BNS', 'NSBH'.
     size : `int`
         number of samples for sampling.
-        default size = 100000.
+        default size = 100000. To get stable rates, size should be large (>=1e6).
     batch_size : `int`
         batch size for SNR calculation.
         default batch_size = 50000.
         reduce the batch size if you are getting memory error.
-        recommended batch_size = 50000, if size = 1000000.
-    snr_finder : `str`
+        recommended batch_size = 200000, if size = 1000000.
+    cosmology : `astropy.cosmology`
+        cosmology to use for the calculation.
+        default cosmology = LambdaCDM(H0=70, Om0=0.3, Ode0=0.7).
+    snr_finder : `str` or `function`
         default snr_finder = 'gwsnr'.
         if None, the SNR will be calculated using the gwsnr package.
-        if 'custom', the SNR will be calculated using a custom function.
-        The custom function should have input and output as given in GWSNR.snr method.
+        if custom snr finder function is provided, the SNR will be calculated using a custom function. The custom function should follow the following signature:
+        def snr_finder(gw_param_dict):
+            ...
+            return optimal_snr_dict
+        where optimal_snr_dict.keys = ['optimal_snr_net']. Refer to `gwsnr` package's GWSNR.snr attribute for more details.
+    pdet_finder : `function`
+        default pdet_finder = None.
+        The rate calculation uses either the pdet_finder or the snr_finder to calculate the detectable events. The custom pdet finder function should follow the following signature:
+        def pdet_finder(gw_param_dict):
+            ...
+            return pdet_net_dict
+        where pdet_net_dict.keys = ['pdet_net']. For example uses, refer to [GRB pdet example](https://ler.readthedocs.io/en/latest/examples/rates/grb%20detection%20rate.html).
+    list_of_detectors : `list`
+        list of detectors.
+        default list_of_detectors = ['H1', 'L1', 'V1']. This is used for lensed SNR calculation wrt to the detectors. Provide 'None' if you only need net SNR/Pdet. Refer to ImageProperties.get_lensed_snrs for more details.
     json_file_names: `dict`
         names of the json files to strore the necessary parameters.
-        default json_file_names = {'ler_params': 'LeR_params.json', 'unlensed_param': 'unlensed_param.json', 'unlensed_param_detectable': 'unlensed_param_detectable.json'}.\n
+        default json_file_names = {'ler_params': 'LeR_params.json', 'unlensed_param': 'unlensed_param.json', 'unlensed_param_detectable': 'unlensed_param_detectable.json'}.
+    interpolator_directory : `str`
+        directory to store the interpolators.
+        default interpolator_directory = './interpolator_pickle'. This is used for storing the various interpolators related to `ler` and `gwsnr` package.
+    ler_directory : `str`
+        directory to store the parameters.
+        default ler_directory = './ler_data'. This is used for storing the parameters of the simulated events.
+    verbose : `bool`
+        default verbose = True.
+        if True, the function will print all chosen parameters.
+        Choose False to prevent anything from printing.
     kwargs : `keyword arguments`
-        Note : kwargs takes input for initializing the :class:`~ler.lens_galaxy_population.LensGalaxyParameterDistribution`, :meth:`~gwsnr_intialization`.
+        Note : kwargs takes input for initializing the :class:`~ler.lens_galaxy_population.LensGalaxyParameterDistribution`, :class:`~ler.gw_source_population.CBCSourceParameterDistribution`, :class:`~ler.gw_source_population.CBCSourceRedshiftDistribution` and :class:`~ler.image_properties.ImageProperties` classes. If snr_finder='gwsnr', then kwargs also takes input for initializing the :class:`~gwsnr.GWSNR` class. Please refer to the respective classes for more details.
 
     Examples
     ----------
@@ -55,6 +84,9 @@ class LeR(LensGalaxyParameterDistribution):
     >>> ler = LeR()
     >>> unlensed_params = ler.unlensed_cbc_statistics();
     >>> ler.unlensed_rate();
+    >>> lensed_params = ler.lensed_cbc_statistics();
+    >>> ler.lensed_rate();
+    >>> ler.rate_ratio();
         
     Instance Attributes
     ----------
@@ -62,6 +94,8 @@ class LeR(LensGalaxyParameterDistribution):
     +-------------------------------------+----------------------------------+
     | Atrributes                          | Type                             |
     +=====================================+==================================+
+    |:attr:`~npool`                       | `int`                            |
+    +-------------------------------------+----------------------------------+
     |:attr:`~z_min`                       | `float`                          |
     +-------------------------------------+----------------------------------+
     |:attr:`~z_max`                       | `float`                          |
@@ -76,7 +110,11 @@ class LeR(LensGalaxyParameterDistribution):
     +-------------------------------------+----------------------------------+
     |:attr:`~json_file_names`             | `dict`                           |
     +-------------------------------------+----------------------------------+
-    |:attr:`~directory`                   | `str`                            |
+    |:attr:`~interpolator_directory`      | `str`                            |
+    +-------------------------------------+----------------------------------+
+    |:attr:`~ler_directory`               | `str`                            |
+    +-------------------------------------+----------------------------------+
+    |:attr:`~gwsnr`                       | `bool`                           |
     +-------------------------------------+----------------------------------+
     |:attr:`~gw_param_sampler_dict`       | `dict`                           |
     +-------------------------------------+----------------------------------+
@@ -108,62 +146,79 @@ class LeR(LensGalaxyParameterDistribution):
     |:meth:`~snr`                         | Function to get the snr with the |
     |                                     | given parameters.                |
     +-------------------------------------+----------------------------------+
+    |:meth:`~snr_bilby`                   | Function to get the snr with the |
+    |                                     | given parameters using inner-    |
+    |                                     | product method.                  |
+    +-------------------------------------+----------------------------------+
+    |:meth:`~pdet`                        | Function to get the pdet with    |
+    |                                     | the given parameters.            |
+    +-------------------------------------+----------------------------------+
     |:meth:`~store_ler_params`            | Function to store the all the    |
     |                                     | necessary parameters.            |
     +-------------------------------------+----------------------------------+
     |:meth:`~unlensed_cbc_statistics`     | Function to generate unlensed    |
-    |                                     | GW source parameters.            |
+    |                                     | GW source parameters in batches. |
     +-------------------------------------+----------------------------------+
     |:meth:`~unlensed_sampling_routine`   | Function to generate unlensed    |
-    |                                     | GW source parameters.            |
+    |                                     | GW source parameters. It stores  |
+    |                                     | the parameters of the generated  |
+    |                                     | events in a json file.           |
     +-------------------------------------+----------------------------------+
     |:meth:`~unlensed_rate`               | Function to calculate the        |
-    |                                     | unlensed rate.                   |
-    +-------------------------------------+----------------------------------+
-    |:meth:`~selecting_n_unlensed_detectable_events`                         |
-    +-------------------------------------+----------------------------------+
-    |                                     | Function to select n unlensed    |
-    |                                     | detectable events.               |
+    |                                     | unlensed rate. It also stores    |
+    |                                     | the parameters of the detectable |
+    |                                     | unlesed events in a json file.   |
     +-------------------------------------+----------------------------------+
     |:meth:`~lensed_cbc_statistics`       | Function to generate lensed      |
     |                                     | GW source parameters.            |
     +-------------------------------------+----------------------------------+
     |:meth:`~lensed_sampling_routine`     | Function to generate lensed      |
-    |                                     | GW source parameters.            |
+    |                                     | GW source parameters. It stores  |
+    |                                     | the parameters of the generated  |
+    |                                     | events in a json file.           |
     +-------------------------------------+----------------------------------+
     |:meth:`~lensed_rate`                 | Function to calculate the        |
-    |                                     | lensed rate.                     |
+    |                                     | lensed rate. It also stores the  |
+    |                                     | parameters of the detectable     |
+    |                                     | lensed events in a json file.    |
     +-------------------------------------+----------------------------------+
     |:meth:`~rate_ratio`                  | Function to calculate the rate   |
-    |                                     | ratio.                           |
+    |                                     | ratio between lensed and         |
+    |                                     | unlensed events.                 |
     +-------------------------------------+----------------------------------+
     |:meth:`~rate_comparision_with_rate_calculation                          |
     +-------------------------------------+----------------------------------+
-    |                                     | Function to compare the rates    |
-    |                                     | calculated using LeR between     |
-    |                                     | unlensed and lensed events.      |
+    |                                     | Function to calculate rates for  |
+    |                                     | unleesed and lensed events and   |
+    |                                     | compare it with the rate. It also|
+    |                                     | stores the parameters of the     |
+    |                                     | detectable events in a json file.|
     +-------------------------------------+----------------------------------+
-    |:meth:`~param_plot`                  | Function to plot the             |
-    |                                     | distribution of various          |
-    |                                     | parameters.                      |
+    |:meth:`~selecting_n_unlensed_detectable_events`                         |
     +-------------------------------------+----------------------------------+
-    |:meth:`~relative_mu_dt_lensed`       | Function to calculate the        |
-    |                                     | relative magnification and       |
-    |                                     | relative time-delay of lensed    |
-    |                                     | events.                          |
+    |                                     | Function to select n unlensed    |
+    |                                     | detectable events. It stores the |
+    |                                     | parameters of the detectable     |
+    |                                     | unlesed events in a json file.   |
     +-------------------------------------+----------------------------------+
-    |:meth:`~relative_mu_dt_unlensed`     | Function to calculate the        |
-    |                                     | relative magnification and       |
-    |                                     | relative time-delay of unlensed  |
-    |                                     | events.                          |
+    |:meth:`~selecting_n_lensed_detectable_events`                           |
     +-------------------------------------+----------------------------------+
-    |:meth:`~ mu_vs_dt_plot`              | Function to plot the             |
-    |                                     | relative magnification vs        |
-    |                                     | relative time-delay.             |
+    |                                     | Function to select n lensed      |
+    |                                     | detectable events. It stores the |
+    |                                     | parameters of the detectable     |
+    |                                     | lensed events in a json file.    |
     +-------------------------------------+----------------------------------+
+
+    Note: `LeR` class also inherits all the instances from the :class:`~ler.lens_galaxy_population.LensGalaxyParameterDistribution` class. Please refer to the :class:`~ler.lens_galaxy_population.LensGalaxyParameterDistribution` class for more details.
     """
 
     # Attributes
+
+    npool = None
+    """``int`` \n
+    Number of logical cores to use.
+    """
+
     z_min = None
     """``float`` \n
     Minimum redshift of the source population
@@ -197,7 +252,7 @@ class LeR(LensGalaxyParameterDistribution):
 
     json_file_names = None
     """``dict`` \n
-    Names of the json files to strore the necessary parameters.
+    Names of the json files to store the necessary parameters.
     """
 
     interpolator_directory = None
@@ -210,14 +265,14 @@ class LeR(LensGalaxyParameterDistribution):
     Directory to store the parameters.
     """
 
-    gw_param_sampler_dict = None
-    """``dict`` \n
-    Dictionary of parameters to initialize the ``CBCSourceParameterDistribution`` class.
+    gwsnr = None
+    """``bool`` \n
+    If True, the SNR will be calculated using the gwsnr package.
     """
 
-    lens_param_sampler_dict = None
+    gw_param_sampler_dict = None
     """``dict`` \n
-    Dictionary of parameters to initialize the ``LensGalaxyParameterDistribution`` class.
+    Dictionary of parameters to initialize the ``CBCSourceParameterDistribution`` class.
     """
 
     snr_calculator_dict = None
@@ -230,6 +285,136 @@ class LeR(LensGalaxyParameterDistribution):
     List of detectors.
     """
 
+    unlensed_param = None
+    """``dict`` \n
+    Dictionary of unlensed GW source parameters. The included parameters and their units are as follows (for default settings):\n
+    +--------------------+--------------+--------------------------------------+
+    | Parameter          | Units        | Description                          |
+    +====================+==============+======================================+
+    | zs                 |              | redshift of the source               |
+    +--------------------+--------------+--------------------------------------+
+    | geocent_time       | s            | GPS time of coalescence              |
+    +--------------------+--------------+--------------------------------------+
+    | ra                 | rad          | right ascension                      |
+    +--------------------+--------------+--------------------------------------+
+    | dec                | rad          | declination                          |
+    +--------------------+--------------+--------------------------------------+
+    | phase              | rad          | phase of GW at reference frequency   |
+    +--------------------+--------------+--------------------------------------+
+    | psi                | rad          | polarization angle                   |
+    +--------------------+--------------+--------------------------------------+
+    | theta_jn           | rad          | inclination angle                    |
+    +--------------------+--------------+--------------------------------------+
+    | luminosity_distance| Mpc          | luminosity distance                  |
+    +--------------------+--------------+--------------------------------------+
+    | mass_1_source      | Msun         | mass_1 of the compact binary         |
+    |                    |              | (source frame)                       |
+    +--------------------+--------------+--------------------------------------+
+    | mass_2_source      | Msun         | mass_2 of the compact binary         |
+    |                    |              | (source frame)                       |
+    +--------------------+--------------+--------------------------------------+
+    | mass_1             | Msun         | mass_1 of the compact binary         |
+    |                    |              | (detector frame)                     |
+    +--------------------+--------------+--------------------------------------+
+    | mass_2             | Msun         | mass_2 of the compact binary         |
+    |                    |              | (detector frame)                     |
+    +--------------------+--------------+--------------------------------------+
+    | L1                 |              | optimal snr of L1                    |
+    +--------------------+--------------+--------------------------------------+
+    | H1                 |              | optimal snr of H1                    |
+    +--------------------+--------------+--------------------------------------+
+    | V1                 |              | optimal snr of V1                    |
+    +--------------------+--------------+--------------------------------------+
+    | optimal_snr_net    |              | optimal snr of the network           |
+    +--------------------+--------------+--------------------------------------+
+    """
+
+    unlensed_param_detectable = None
+    """``dict`` \n
+    Dictionary of detectable unlensed GW source parameters. It includes the same parameters as the :attr:`~unlensed_param` attribute.
+    """
+
+    lensed_param = None
+    """``dict`` \n
+    Dictionary of lens parameters, images parameters and lensed GW source parameters. The included parameters and their units are as follows (for default settings):\n
+    +------------------------------+-----------+-------------------------------+
+    | Parameter                    | Units     | Description                   |
+    +==============================+===========+===============================+
+    | zl                           |           | redshift of the lens          |
+    +------------------------------+-----------+-------------------------------+
+    | zs                           |           | redshift of the source        |
+    +------------------------------+-----------+-------------------------------+
+    | sigma                        |km s^-1    | velocity dispersion           |
+    +------------------------------+-----------+-------------------------------+
+    | q                            |           | axis ratio                    |
+    +------------------------------+-----------+-------------------------------+
+    | theta_E                      | arcsec    | Einstein radius               |
+    +------------------------------+-----------+-------------------------------+
+    | phi                          | rad       | axis rotation angle           |
+    +------------------------------+-----------+-------------------------------+
+    | e1                           |           | ellipticity component 1       |
+    +------------------------------+-----------+-------------------------------+
+    | e2                           |           | ellipticity component 2       |
+    +------------------------------+-----------+-------------------------------+
+    | gamma1                       |           | shear component 1             |
+    +------------------------------+-----------+-------------------------------+
+    | gamma2                       |           | shear component 2             |
+    +------------------------------+-----------+-------------------------------+
+    | gamma                        |           | shear                         |
+    +------------------------------+-----------+-------------------------------+
+    | ra                           | rad       | right ascension               |
+    +------------------------------+-----------+-------------------------------+
+    | dec                          | rad       | declination                   |
+    +------------------------------+-----------+-------------------------------+
+    | phase                        | rad       | phase of GW at reference freq |
+    +------------------------------+-----------+-------------------------------+
+    | psi                          | rad       | polarization angle            |
+    +------------------------------+-----------+-------------------------------+
+    | theta_jn                     | rad       | inclination angle             |
+    +------------------------------+-----------+-------------------------------+
+    | mass_1_source                | Msun      | mass_1 of the compact binary  |
+    |                              |           | (source frame)                |
+    +------------------------------+-----------+-------------------------------+
+    | mass_2_source                | Msun      | mass_2 of the compact binary  |
+    |                              |           | (source frame)                |
+    +------------------------------+-----------+-------------------------------+
+    | mass_1                       | Msun      | mass_1 of the compact binary  |
+    |                              |           | (detector frame)              |
+    +------------------------------+-----------+-------------------------------+
+    | mass_2                       | Msun      | mass_2 of the compact binary  |
+    |                              |           | (detector frame)              |
+    +------------------------------+-----------+-------------------------------+
+    | x0_image_positions           |           | x0 image positions            |
+    +------------------------------+-----------+-------------------------------+
+    | x1_image_positions           |           | x1 image positions            |
+    +------------------------------+-----------+-------------------------------+
+    | magnifications               |           | magnifications                |
+    +------------------------------+-----------+-------------------------------+
+    | time_delays                  |           | time delays                   |
+    +------------------------------+-----------+-------------------------------+
+    | image_type                   |           | image type                    |
+    +------------------------------+-----------+-------------------------------+
+    | n_images                     |           | number of images              |
+    +------------------------------+-----------+-------------------------------+
+    | effective_luminosity_distance| Mpc       | effective luminosity distance |
+    +------------------------------+-----------+-------------------------------+
+    | effective_geocent_time       | s         | effective GPS time of coalesc |
+    +------------------------------+-----------+-------------------------------+
+    | L1                           |           | optimal snr of L1             |
+    +------------------------------+-----------+-------------------------------+
+    | H1                           |           | optimal snr of H1             |
+    +------------------------------+-----------+-------------------------------+
+    | V1                           |           | optimal snr of V1             |
+    +------------------------------+-----------+-------------------------------+
+    | optimal_snr_net              |           | optimal snr of the network    |
+    +------------------------------+-----------+-------------------------------+
+    """
+
+    lensed_param_detectable = None
+    """``dict`` \n
+    Dictionary of detectable lensed GW source parameters.
+    """
+
     def __init__(
         self,
         npool=int(4),
@@ -294,114 +479,113 @@ class LeR(LensGalaxyParameterDistribution):
         """
         Function to print all the parameters.
         """
-
         # print all relevant functions and sampler priors
         print("\n LeR set up params:")
-        print("npool = ", self.npool)
-        print("z_min = ", self.z_min)
-        print("z_max = ", self.z_max)
-        print("event_type = ", self.event_type)
-        print("size = ", self.size)
-        print("batch_size = ", self.batch_size)
-        print("cosmology = ", self.cosmo)
+        print(f'npool = {self.npool},')
+        print(f'z_min = {self.z_min},')
+        print(f'z_max = {self.z_max},')
+        print(f"event_type = '{self.event_type}',")
+        print(f'size = {self.size},')
+        print(f'batch_size = {self.batch_size},')
+        print(f'cosmology = {self.cosmo},')
         if self.snr:
-            print("snr_finder = ", self.snr)
+            print(f'snr_finder = {self.snr},')
         if self.pdet:
-            print("pdet_finder = ", self.pdet)
-        print("json_file_names = ", self.json_file_names)
-        print("interpolator_directory = ", self.interpolator_directory)
-        print("ler_directory = ", self.ler_directory)
+            print(f'pdet_finder = {self.pdet},')
+        print(f'json_file_names = {self.json_file_names},')
+        print(f'interpolator_directory = {self.interpolator_directory},')
+        print(f'ler_directory = {self.ler_directory},')
+
+        print("\n LeR also takes CBCSourceParameterDistribution class params as kwargs, as follows:")
+        print(f"source_priors = {self.gw_param_sampler_dict['source_priors']},")
+        print(f"source_priors_params = {self.gw_param_sampler_dict['source_priors_params']},")
+        print(f"spin_zero = {self.gw_param_sampler_dict['spin_zero']},")
+        print(f"spin_precession = {self.gw_param_sampler_dict['spin_precession']},")
+        print(f"create_new_interpolator = {self.gw_param_sampler_dict['create_new_interpolator']},")
+        
+        print("\n LeR also takes LensGalaxyParameterDistribution class params as kwargs, as follows:")
+        print(f"lens_type = '{self.gw_param_sampler_dict['lens_type']}',")
+        print(f"lens_functions = {self.gw_param_sampler_dict['lens_functions']},")
+        print(f"lens_priors = {self.gw_param_sampler_dict['lens_priors']},")
+        print(f"lens_priors_params = {self.gw_param_sampler_dict['lens_priors_params']},")
+        
+        print("\n LeR also takes ImageProperties class params as kwargs, as follows:")
+        print(f"n_min_images = {self.n_min_images},")
+        print(f"n_max_images = {self.n_max_images},")
+        print(f"geocent_time_min = {self.geocent_time_min},")
+        print(f"geocent_time_max = {self.geocent_time_max},")
+        print(f"lens_model_list = {self.lens_model_list},")
         
-        print("\n LeR also takes CBCSourceParameterDistribution params as kwargs, as follows:")
-        print("source_priors=", self.gw_param_sampler_dict["source_priors"])
-        print("source_priors_params=", self.gw_param_sampler_dict["source_priors_params"])
-        print("spin_zero=", self.gw_param_sampler_dict["spin_zero"])
-        print("spin_precession=", self.gw_param_sampler_dict["spin_precession"])
-        print("create_new_interpolator=", self.gw_param_sampler_dict["create_new_interpolator"])
-
-        print("\n LeR also takes LensGalaxyParameterDistribution params as kwargs, as follows:")
-        print("lens_type = ", self.gw_param_sampler_dict["lens_type"])
-        print("lens_functions = ", self.gw_param_sampler_dict["lens_functions"])
-        print("lens_priors = ", self.gw_param_sampler_dict["lens_priors"])
-        print("lens_priors_params = ", self.gw_param_sampler_dict["lens_priors_params"])
-
-        print("\n Image properties:")
-        print("n_min_images = ", self.n_min_images)
-        print("n_max_images = ", self.n_max_images)
-        print("geocent_time_min = ", self.geocent_time_min)
-        print("geocent_time_max = ", self.geocent_time_max)
-        print("lens_model_list = ", self.lens_model_list)
-
         if self.gwsnr:
             print("\n LeR also takes gwsnr.GWSNR params as kwargs, as follows:")
-            print("mtot_min = ", self.snr_calculator_dict["mtot_min"])
-            print("mtot_max = ", self.snr_calculator_dict["mtot_max"])
-            print("ratio_min = ", self.snr_calculator_dict["ratio_min"])
-            print("ratio_max = ", self.snr_calculator_dict["ratio_max"])
-            print("mtot_resolution = ", self.snr_calculator_dict["mtot_resolution"])
-            print("ratio_resolution = ", self.snr_calculator_dict["ratio_resolution"])
-            print("sampling_frequency = ", self.snr_calculator_dict["sampling_frequency"])
-            print("waveform_approximant = ", self.snr_calculator_dict["waveform_approximant"])
-            print("minimum_frequency = ", self.snr_calculator_dict["minimum_frequency"])
-            print("snr_type = ", self.snr_calculator_dict["snr_type"])
-            print("psds = ", self.snr_calculator_dict["psds"])
-            print("ifos = ", self.snr_calculator_dict["ifos"])
-            print("interpolator_dir = ", self.snr_calculator_dict["interpolator_dir"])
-            print("create_new_interpolator = ", self.snr_calculator_dict["create_new_interpolator"])
-            print("gwsnr_verbose = ", self.snr_calculator_dict["gwsnr_verbose"])
-            print("multiprocessing_verbose = ", self.snr_calculator_dict["multiprocessing_verbose"])
-            print("mtot_cut = ", self.snr_calculator_dict["mtot_cut"])
-        del self.gwsnr
+            print(f"mtot_min = {self.snr_calculator_dict['mtot_min']},")
+            print(f"mtot_max = {self.snr_calculator_dict['mtot_max']},")
+            print(f"ratio_min = {self.snr_calculator_dict['ratio_min']},")
+            print(f"ratio_max = {self.snr_calculator_dict['ratio_max']},")
+            print(f"mtot_resolution = {self.snr_calculator_dict['mtot_resolution']},")
+            print(f"ratio_resolution = {self.snr_calculator_dict['ratio_resolution']},")
+            print(f"sampling_frequency = {self.snr_calculator_dict['sampling_frequency']},")
+            print(f"waveform_approximant = '{self.snr_calculator_dict['waveform_approximant']}',")
+            print(f"minimum_frequency = {self.snr_calculator_dict['minimum_frequency']},")
+            print(f"snr_type = '{self.snr_calculator_dict['snr_type']}',")
+            print(f"psds = {self.snr_calculator_dict['psds']},")
+            print(f"ifos = {self.snr_calculator_dict['ifos']},")
+            print(f"interpolator_dir = '{self.snr_calculator_dict['interpolator_dir']}',")
+            print(f"create_new_interpolator = {self.snr_calculator_dict['create_new_interpolator']},")
+            print(f"gwsnr_verbose = {self.snr_calculator_dict['gwsnr_verbose']},")
+            print(f"multiprocessing_verbose = {self.snr_calculator_dict['multiprocessing_verbose']},")
+            print(f"mtot_cut = {self.snr_calculator_dict['mtot_cut']},")
+        # del self.gwsnr
 
         print("\n For reference, the chosen source parameters are listed below:")
-        print("merger_rate_density = ", self.gw_param_samplers["merger_rate_density"])
+        print(f"merger_rate_density = '{self.gw_param_samplers['merger_rate_density']}'")
         print("merger_rate_density_params = ", self.gw_param_samplers_params["merger_rate_density"])
-        print("source_frame_masses = ", self.gw_param_samplers["source_frame_masses"])
+        print(f"source_frame_masses = '{self.gw_param_samplers['source_frame_masses']}'")
         print("source_frame_masses_params = ", self.gw_param_samplers_params["source_frame_masses"])
-        print("geocent_time = ", self.gw_param_samplers["geocent_time"])
+        print(f"geocent_time = '{self.gw_param_samplers['geocent_time']}'")
         print("geocent_time_params = ", self.gw_param_samplers_params["geocent_time"])
-        print("ra = ", self.gw_param_samplers["ra"])
+        print(f"ra = '{self.gw_param_samplers['ra']}'")
         print("ra_params = ", self.gw_param_samplers_params["ra"])
-        print("dec = ", self.gw_param_samplers["dec"])
+        print(f"dec = '{self.gw_param_samplers['dec']}'")
         print("dec_params = ", self.gw_param_samplers_params["dec"])
-        print("phase = ", self.gw_param_samplers["phase"])
+        print(f"phase = '{self.gw_param_samplers['phase']}'")
         print("phase_params = ", self.gw_param_samplers_params["phase"])
-        print("psi = ", self.gw_param_samplers["psi"])
+        print(f"psi = '{self.gw_param_samplers['psi']}'")
         print("psi_params = ", self.gw_param_samplers_params["psi"])
-        print("theta_jn = ", self.gw_param_samplers["theta_jn"])
+        print(f"theta_jn = '{self.gw_param_samplers['theta_jn']}'")
         print("theta_jn_params = ", self.gw_param_samplers_params["theta_jn"])
-        if self.spin_zero==False:
-            print("a_1 = ", self.gw_param_samplers["a_1"])
+        if self.spin_zero is False:
+            print(f"a_1 = '{self.gw_param_samplers['a_1']}'")
             print("a_1_params = ", self.gw_param_samplers_params["a_1"])
-            print("a_2 = ", self.gw_param_samplers["a_2"])
+            print(f"a_2 = '{self.gw_param_samplers['a_2']}'")
             print("a_2_params = ", self.gw_param_samplers_params["a_2"])
-            if self.spin_precession==True:
-                print("tilt_1 = ", self.gw_param_samplers["tilt_1"])
+            if self.spin_precession is True:
+                print(f"tilt_1 = '{self.gw_param_samplers['tilt_1']}'")
                 print("tilt_1_params = ", self.gw_param_samplers_params["tilt_1"])
-                print("tilt_2 = ", self.gw_param_samplers["tilt_2"])
+                print(f"tilt_2 = '{self.gw_param_samplers['tilt_2']}'")
                 print("tilt_2_params = ", self.gw_param_samplers_params["tilt_2"])
-                print("phi_12 = ", self.gw_param_samplers["phi_12"])
+                print(f"phi_12 = '{self.gw_param_samplers['phi_12']}'")
                 print("phi_12_params = ", self.gw_param_samplers_params["phi_12"])
-                print("phi_jl = ", self.gw_param_samplers["phi_jl"])
+                print(f"phi_jl = '{self.gw_param_samplers['phi_jl']}'")
                 print("phi_jl_params = ", self.gw_param_samplers_params["phi_jl"])
 
         print("\n For reference, the chosen lens related parameters and functions are listed below:")
-        print("lens_redshift = ", self.lens_param_samplers["lens_redshift"])
+        print(f"lens_redshift = '{self.lens_param_samplers['lens_redshift']}'")
         print("lens_redshift_params = ", self.lens_param_samplers_params["lens_redshift"])
-        print("velocity_dispersion = ", self.lens_param_samplers["velocity_dispersion"])
+        print(f"velocity_dispersion = '{self.lens_param_samplers['velocity_dispersion']}'")
         print("velocity_dispersion_params = ", self.lens_param_samplers_params["velocity_dispersion"])
-        print("axis_ratio = ", self.lens_param_samplers["axis_ratio"])
+        print(f"axis_ratio = '{self.lens_param_samplers['axis_ratio']}'")
         print("axis_ratio_params = ", self.lens_param_samplers_params["axis_ratio"])
-        print("axis_rotation_angle = ", self.lens_param_samplers["axis_rotation_angle"])
+        print(f"axis_rotation_angle = '{self.lens_param_samplers['axis_rotation_angle']}'")
         print("axis_rotation_angle_params = ", self.lens_param_samplers_params["axis_rotation_angle"])
-        print("shear = ", self.lens_param_samplers["shear"])
+        print(f"shear = '{self.lens_param_samplers['shear']}'")
         print("shear_params = ", self.lens_param_samplers_params["shear"])
-        print("mass_density_spectral_index = ", self.lens_param_samplers["mass_density_spectral_index"])
+        print(f"mass_density_spectral_index = '{self.lens_param_samplers['mass_density_spectral_index']}'")
         print("mass_density_spectral_index_params = ", self.lens_param_samplers_params["mass_density_spectral_index"])
         # lens functions
         print("Lens functions:")
-        print("strong_lensing_condition = ", self.lens_functions["strong_lensing_condition"])
-        print("optical_depth = ", self.lens_functions["optical_depth"])
+        print(f"strong_lensing_condition = '{self.lens_functions['strong_lensing_condition']}'")
+        print(f"optical_depth = '{self.lens_functions['optical_depth']}'")
 
     @property
     def snr(self):
@@ -517,14 +701,7 @@ class LeR(LensGalaxyParameterDistribution):
 
     def class_initialization(self, params=None):
         """
-        Function to initialize the parent classes. List of relevant initialized instances, \n
-        1. self.sample_source_redshift
-        2. self.sample_unlensed_parameters
-        3. self.normalization_pdf_z
-        4. self.sample_lens_parameters
-        5. self.normalization_pdf_z_lensed
-        6. self.image_properties
-        7. self.get_lensed_snrs
+        Function to initialize the parent classes. 
 
         Parameters
         ----------
@@ -586,7 +763,7 @@ class LeR(LensGalaxyParameterDistribution):
 
     def gwsnr_intialization(self, params=None):
         """
-        Function to initialize the gwsnr class
+        Function to initialize the GWSNR class from the `gwsnr` package.
 
         Parameters
         ----------
@@ -612,7 +789,7 @@ class LeR(LensGalaxyParameterDistribution):
             ifos=None,
             interpolator_dir=self.interpolator_directory,
             create_new_interpolator=False,
-            gwsnr_verbose=True,
+            gwsnr_verbose=False,
             multiprocessing_verbose=True,
             mtot_cut=True,
         )
@@ -649,7 +826,7 @@ class LeR(LensGalaxyParameterDistribution):
         self.snr_bilby = gwsnr.compute_bilby_snr
         self.snr_calculator_dict["mtot_max"] = gwsnr.mtot_max
         self.snr_calculator_dict["psds"] = gwsnr.psds_list
-        # self.pdet = gwsnr.pdet
+        #self.pdet = gwsnr.pdet
 
     def store_ler_params(self, output_jsonfile="ler_params.json"):
         """
@@ -698,7 +875,7 @@ class LeR(LensGalaxyParameterDistribution):
         self, size=None, resume=False, save_batch=False, output_jsonfile=None,
     ):
         """
-        Function to generate unlensed GW source parameters. This function also stores the parameters in json file.
+        Function to generate unlensed GW source parameters. This function calls the unlensed_sampling_routine function to generate the parameters in batches. The generated parameters are stored in a json file; and if save_batch=True, it keeps updating the file in batches.
 
         Parameters
         ----------
@@ -712,13 +889,12 @@ class LeR(LensGalaxyParameterDistribution):
             if True, the function will save the parameters in batches. if False, the function will save all the parameters at the end of sampling. save_batch=False is faster.
         output_jsonfile : `str`
             json file name for storing the parameters.
-            default output_jsonfile = 'unlensed_params.json'.
+            default output_jsonfile = 'unlensed_params.json'. Note that this file will be stored in the self.ler_directory.
 
         Returns
         ----------
         unlensed_param : `dict`
-            dictionary of unlensed GW source parameters.
-            unlensed_param.keys() = ['zs', 'geocent_time', 'ra', 'dec', 'phase', 'psi', 'theta_jn', 'luminosity_distance', 'mass_1_source', 'mass_2_source', 'mass_1', 'mass_2', 'optimal_snr_net', 'L1', 'H1', 'V1']
+            dictionary of unlensed GW source parameters. Refer to :attr:`~unlensed_param` for details.
 
         Examples
         ----------
@@ -756,7 +932,7 @@ class LeR(LensGalaxyParameterDistribution):
             if self.dict_buffer:
                 unlensed_param = self.dict_buffer.copy()
                 # store all params in json file
-                print(f"saving all unlensed_params in {output_path}...")
+                print(f"saving all unlensed_params in {output_path} ")
                 append_json(output_path, unlensed_param, replace=True)
             else:
                 print("unlensed_params already sampled.")
@@ -767,25 +943,26 @@ class LeR(LensGalaxyParameterDistribution):
     
     def unlensed_sampling_routine(self, size, output_jsonfile, resume=False, save_batch=True):
         """
-        Function to generate unlensed GW source parameters. This function also stores the parameters in json file.
+        Function to generate unlensed GW source parameters. This function also stores the parameters in json file in the current batch if save_batch=True.
 
         Parameters
         ----------
         size : `int`
             number of samples.
             default size = 100000.
-        resume : `bool`
-            resume = False (default) or True.
-            if True, the function will resume from the last batch.
         output_jsonfile : `str`
             json file name for storing the parameters.
-            default output_jsonfile = 'unlensed_params.json'.
+            default output_jsonfile = 'unlensed_params.json'. Note that this file will be stored in the self.ler_directory.
+        resume : `bool`
+            resume = False (default) or True. 
+            if True, it appends the new samples to the existing json file.
+        save_batch : `bool`
+            if True, the function will save the parameters in batches. if False, the function will save all the parameters at the end of sampling. save_batch=False is faster.
 
         Returns
         ----------
         unlensed_param : `dict`
-            dictionary of unlensed GW source parameters.
-            unlensed_param.keys() = ['zs', 'geocent_time', 'ra', 'dec', 'phase', 'psi', 'theta_jn', 'luminosity_distance', 'mass_1_source', 'mass_2_source', 'mass_1', 'mass_2', 'optimal_snr_net', 'L1', 'H1', 'V1']
+            dictionary of unlensed GW source parameters. Refer to :attr:`~unlensed_param` for details.
         """
 
         # get gw params
@@ -818,13 +995,19 @@ class LeR(LensGalaxyParameterDistribution):
         self,
         unlensed_param=None,
         snr_threshold=8.0,
+        pdet_threshold=0.5,
         output_jsonfile=None,
         detectability_condition="step_function",
         snr_recalculation=False,
-        threshold_snr_recalculation=6.0,
+        snr_threshold_recalculation=[4, 20],
     ):
         """
-        Function to calculate the unlensed rate. This function also stores the parameters of the detectable events in json file.
+        Function to calculate the unlensed rate. This function also stores the parameters of the detectable events in json file. There are two conditions for detectability: 'step_function' and 'pdet'.
+
+        1. 'step_function': If two images have SNR>8.0, then the event is detectable. This is a step function. This is with the assumption that SNR function is provided and not None. 
+        2. 'pdet':
+            i) If self.pdet is None and self.snr is not None, then it will calculate the pdet from the snr. There is no hard cut for this pdet and can have value ranging from 0 to 1 near the threshold.
+            ii) If self.pdet is not None, then it will use the generated pdet.
 
         Parameters
         ----------
@@ -834,6 +1017,9 @@ class LeR(LensGalaxyParameterDistribution):
         snr_threshold : `float`
             threshold for detection signal to noise ratio.
             e.g. snr_threshold = 8.
+        pdet_threshold : `float`
+            threshold for detection probability.
+            e.g. pdet_threshold = 0.5.
         output_jsonfile : `str`
             json file name for storing the parameters of the detectable events.
             default output_jsonfile = 'unlensed_params_detectable.json'.
@@ -842,18 +1028,18 @@ class LeR(LensGalaxyParameterDistribution):
             default detectability_condition = 'step_function'.
             other options are 'pdet'.
         snr_recalculation : `bool`
-            if True, the SNR of centain events (snr>threshold_snr_recalculation)will be recalculate with 'inner product'. This is useful when the snr is calculated with 'ann' method.
+            if True, the SNR of centain events (snr>snr_threshold_recalculation)will be recalculate with 'inner-product' method. This is useful when the snr is calculated with 'ann' method of `gwsnr`.
             default snr_recalculation = False.
-        threshold_snr_recalculation : `float`
-            threshold for recalculation of detection signal to noise ratio.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+            default snr_threshold_recalculation = [4, 20].
 
         Returns
         ----------
         total_rate : `float`
             total unlensed rate (Mpc^-3 yr^-1).
         unlensed_param : `dict`
-            dictionary of unlensed GW source parameters of the detectable events.
-            unlensed_param.keys() = ['zs', 'geocent_time', 'ra', 'dec', 'phase', 'psi', 'theta_jn', 'luminosity_distance', 'mass_1_source', 'mass_2_source', 'mass_1', 'mass_2', 'optimal_snr_net', 'L1', 'H1', 'V1']
+            dictionary of unlensed GW source parameters of the detectable events. Refer to :attr:`~unlensed_param` for details.
 
         Examples
         ----------
@@ -863,39 +1049,112 @@ class LeR(LensGalaxyParameterDistribution):
         >>> total_rate, unlensed_param_detectable = ler.unlensed_rate()
         """
         
-        # call self.json_file_names["ler_param"] and for adding the final results
-        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
+        unlensed_param = self._load_param(unlensed_param, param_type="unlensed")
+        total_events = len(unlensed_param["zs"])
+
+        # below is use when the snr is calculated with 'ann' method of `gwsnr`
+        if snr_recalculation:
+            unlensed_param = self._recalculate_snr_unlensed(unlensed_param, snr_threshold_recalculation)
+
+        # find index of detectable events
+        idx_detectable = self._find_detectable_index_unlensed(unlensed_param, snr_threshold, pdet_threshold, detectability_condition)
+
+        detectable_events = np.sum(idx_detectable)
+        # montecarlo integration
+        # The total rate R = norm <Theta(rho-rhoc)>
+        total_rate = self.rate_function(detectable_events, total_events, param_type="unlensed")
+
+        # store all detectable params in json file
+        self._save_detectable_params(output_jsonfile, unlensed_param, idx_detectable, key_file_name="unlensed_param_detectable", nan_to_num=False, verbose=True, replace_jsonfile=True)
+
+        # append ler_param and save it
+        self._append_ler_param(total_rate, detectability_condition)
+        
+        return total_rate, unlensed_param
+
+    def _load_param(self, param, param_type="unlensed"):
+        """
+        Helper function to load or copy unlensed parameters.
         
-        # get gw params from json file if not provided
-        if unlensed_param is None:
-            unlensed_param = self.json_file_names["unlensed_param"]
-        if type(unlensed_param) == str:
-            self.json_file_names["unlensed_param"] = unlensed_param
-            path_ = self.ler_directory+"/"+unlensed_param
-            print(f"getting unlensed_params from json file {path_}...")
-            unlensed_param = get_param_from_json(self.ler_directory+"/"+unlensed_param)
+        Parameters
+        ----------
+        param : `dict` or `str`
+            dictionary of unlensed/lensed parameters or json file name.
+        param_type : `str`
+            type of parameters.
+            default param_type = 'unlensed'. Other options is 'lensed'.
+
+        Returns
+        ----------
+        param : `dict`
+            dictionary of unlensed/lensed parameters.
+        """
+
+        param_type = param_type+"_param"
+        if param is None:
+            param = self.json_file_names[param_type]
+        if isinstance(param, str):
+            path_ = self.ler_directory + "/" + param
+            print(f"Getting {param_type} from json file {path_}...")
+            return get_param_from_json(path_)
         else:
-            print("using provided unlensed_param dict...")
-            unlensed_param = unlensed_param.copy()
+            print("Using provided {param_type} dict...")
+            return param.copy()
 
-        # recalculate snr if required
-        # this ensures that the snr is recalculated for the detectable events
-        # with inner product
-        total_events = len(unlensed_param["zs"])
-        if snr_recalculation:
-            # select only above centain snr threshold
-            param = unlensed_param["optimal_snr_net"]
-            idx_detectable = param > threshold_snr_recalculation
-            # reduce the size of the dict
-            for key, value in unlensed_param.items():
-                unlensed_param[key] = value[idx_detectable]
-            # recalculate more accurate snrs 
-            snrs = self.snr_bilby(gw_param_dict=unlensed_param)
-            unlensed_param.update(snrs)
+    def _recalculate_snr_unlensed(self, unlensed_param, snr_threshold_recalculation):
+        """
+        Recalculates SNR for events where the initial SNR is above a given threshold.
+        
+        Parameters
+        ---------- 
+        unlensed_param : `dict`
+            dictionary of unlensed GW source parameters.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+            default snr_threshold_recalculation = [4, 20].
+
+        Returns
+        ----------
+        unlensed_param : `dict`
+            dictionary of unlensed GW source parameters.
+        """
+
+        snr_param = unlensed_param["optimal_snr_net"]
+        idx_detectable = (snr_param > snr_threshold_recalculation[0]) & (snr_param < snr_threshold_recalculation[1])
+        # reduce the size of the dict
+        for key, value in unlensed_param.items():
+            unlensed_param[key] = value[idx_detectable]
+        # recalculate more accurate snrs 
+        snrs = self.snr_bilby(gw_param_dict=unlensed_param)
+        unlensed_param.update(snrs)
+        return unlensed_param
+
+    def _find_detectable_index_unlensed(self, unlensed_param, snr_threshold, pdet_threshold, detectability_condition):
+        """
+        Find the index of detectable events based on SNR or p_det.
+        
+        Parameters
+        ----------
+        unlensed_param : `dict`
+            dictionary of unlensed GW source parameters.
+        snr_threshold : `float`
+            threshold for detection signal to noise ratio.
+        pdet_threshold : `float`
+            threshold for detection probability.
+        detectability_condition : `str`
+            detectability condition.
+            default detectability_condition = 'step_function'.
+            other options are 'pdet'.
+
+        Returns
+        ----------
+        idx_detectable : `numpy.ndarray`
+            index of detectable events.
+        """
 
         if self.snr:
             if "optimal_snr_net" not in unlensed_param:
-                raise ValueError("'optimal_snr_net' not in unlensed parm dict provided")
+                raise ValueError("'optimal_snr_net' not in unlensed param dict provided")
             if detectability_condition == "step_function":
                 print("given detectability_condition == 'step_function'")
                 param = unlensed_param["optimal_snr_net"]
@@ -904,55 +1163,141 @@ class LeR(LensGalaxyParameterDistribution):
                 print("given detectability_condition == 'pdet'")
                 param = 1 - norm.cdf(snr_threshold - unlensed_param["optimal_snr_net"])
                 unlensed_param["pdet_net"] = param
-                threshold = 0.5
+                threshold = pdet_threshold
         elif self.pdet:
             if "pdet_net" in unlensed_param:
                 print("given detectability_condition == 'pdet'")
                 param = unlensed_param["pdet_net"]
-                threshold = 0.5
+                threshold = pdet_threshold
             else:
-                raise ValueError("'pdet_net' not in unlensed parm dict provided")
+                raise ValueError("'pdet_net' not in unlensed param dict provided")
 
         idx_detectable = param > threshold
-        detectable_events = np.sum(idx_detectable)
-        # montecarlo integration
-        # The total rate R = norm <Theta(rho-rhoc)>
-        total_rate = self.normalization_pdf_z * detectable_events/total_events
-        print(f"total unlensed rate (yr^-1) (with step function): {total_rate}")
-        print(f"number of simulated unlensed detectable events: {detectable_events}")
-        print(f"number of all simulated unlensed events: {total_events}")
+        return idx_detectable
+
+    def rate_function(self, detectable_size, total_size, param_type="unlensed", verbose=True):
+        """
+        General helper function to calculate the rate for unlensed and lensed events.
+
+        Parameters
+        ----------
+        detectable_size : `int`
+            number of detectable events.
+        total_size : `int`
+            total number of events.
+        param_type : `str`
+            type of parameters.
+
+        Returns
+        ----------
+        rate : `float`
+            rate of the events.
+
+        Examples
+        ----------
+        >>> from ler.rates import LeR
+        >>> ler = LeR()
+        >>> rate = ler.rate_function(detectable_size=100, total_size=1000)
+        """
+
+        if param_type == "unlensed":
+            normalization = self.normalization_pdf_z
+        elif param_type == "lensed":
+            normalization = self.normalization_pdf_z_lensed
+        rate = normalization * detectable_size / total_size
+
+        if verbose:
+            print(f"total {param_type} rate (yr^-1): {rate}")
+            print(f"number of simulated {param_type} detectable events: {detectable_size}")
+            print(f"number of simulated all {param_type} events: {total_size}")
+
+        return rate
+
+    def _save_detectable_params(self,
+        output_jsonfile,
+        param, 
+        idx_detectable, 
+        key_file_name="unlensed_param_detectable",
+        nan_to_num=False,
+        verbose=True,
+        replace_jsonfile=True,
+    ):
+        """
+        Helper function to save the detectable parameters in json file.
+
+        Parameters
+        ----------
+        output_jsonfile : `str`
+            json file name for storing the parameters of the detectable events. This is stored in the self.ler_directory.
+        param : `dict`
+            dictionary of GW source parameters.
+        idx_detectable : `numpy.ndarray`
+            index of detectable events.
+        key_file_name : `str`
+            key name for the json file to be added in self.json_file_names.
+        nan_to_num : `bool`
+            if True, it will replace nan with 0.
+            default nan_to_num = False.
+        verbose : `bool`
+            if True, it will print the path of the json file.
+            default verbose = True.
+        replace_jsonfile : `bool`
+            if True, it will replace the json file. If False, it will append the json file.
+        """
 
         # store all detectable params in json file
-        for key, value in unlensed_param.items():
-            unlensed_param[key] = value[idx_detectable]
+        if nan_to_num:
+            for key, value in param.items():
+                param[key] = np.nan_to_num(value[idx_detectable])
+        else:
+            for key, value in param.items():
+                param[key] = value[idx_detectable]
 
         # store all detectable params in json file
         if output_jsonfile is None:
-            output_jsonfile = self.json_file_names["unlensed_param_detectable"]
+            output_jsonfile = self.json_file_names[key_file_name]
         else:
-            self.json_file_names["unlensed_param_detectable"] = output_jsonfile
-        path_ = self.ler_directory+"/"+output_jsonfile
-        print(f"storing detectable unlensed params in {path_}")
-        append_json(self.ler_directory+"/"+output_jsonfile, unlensed_param, replace=True)
+            self.json_file_names[key_file_name] = output_jsonfile
+
+        output_path = self.ler_directory+"/"+output_jsonfile
+        if verbose:
+            print(f"storing detectable params in {output_path}")
+        append_json(output_path, param, replace=replace_jsonfile)
+
+    def _append_ler_param(self, total_rate, detectability_condition, param_type="unlensed"):
+        """
+        Helper function to append the final results, total_rate, in the json file.
 
+        Parameters
+        ----------
+        total_rate : `float`
+            total rate.
+        detectability_condition : `str`
+            detectability condition.
+        param_type : `str`
+            type of parameters.
+            default param_type = 'unlensed'. Other options is 'lensed'.
+        """
+
+        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
         # write the results
-        data["detectable_unlensed_rate_per_year"] = total_rate
-        data["detectability_condition"] = detectability_condition
+        data[f"detectable_{param_type}_rate_per_year"] = total_rate
+        data[f"detectability_condition+{param_type}"] = detectability_condition
         append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
-        
-        return total_rate, unlensed_param
     
     def lensed_cbc_statistics(
         self, size=None, save_batch=False, resume=False, output_jsonfile=None,
     ):
         """
-        Function to generate lensed GW source parameters. This function also stores the parameters in json file.
+        Function to generate lensed GW source parameters. This function calls the lensed_sampling_routine function to generate the parameters in batches. The generated parameters are stored in a json file; and if save_batch=True, it keeps updating the file in batches.
 
         Parameters
         ----------
         size : `int`
             number of samples.
             default size = 100000.
+        save_batch : `bool`
+            if True, the function will save the parameters in batches. if False, the function will save all the parameters at the end of sampling. save_batch=False is faster.
         resume : `bool`
             resume = False (default) or True.
             if True, the function will resume from the last batch.
@@ -963,8 +1308,7 @@ class LeR(LensGalaxyParameterDistribution):
         Returns
         ----------
         lensed_param : `dict`
-            dictionary of lensed GW source parameters.
-            lensed_param.keys() = 
+            dictionary of lensed GW source parameters. Refer to :attr:`~lensed_param` for details.
 
         Examples
         ----------
@@ -1002,7 +1346,7 @@ class LeR(LensGalaxyParameterDistribution):
             if self.dict_buffer:
                 lensed_param = self.dict_buffer.copy()
                 # store all params in json file
-                print(f"saving all lensed_params in {output_path}...")
+                print(f"saving all lensed_params in {output_path} ")
                 append_json(output_path, lensed_param, replace=True)
             else:
                 print("lensed_params already sampled.")
@@ -1013,25 +1357,26 @@ class LeR(LensGalaxyParameterDistribution):
     
     def lensed_sampling_routine(self, size, output_jsonfile, save_batch=True, resume=False):
         """
-        Function to generate lensed GW source parameters. This function also stores the parameters in json file.
+        Function to generate lensed GW source parameters. This function also stores the parameters in json file in the current batch if save_batch=True.
 
         Parameters
         ----------
         size : `int`
             number of samples.
             default size = 100000.
-        resume : `bool`
-            resume = False (default) or True.
-            if True, the function will resume from the last batch.
         output_jsonfile : `str`
             json file name for storing the parameters.
-            default output_jsonfile = 'lensed_params.json'.
+            default output_jsonfile = 'lensed_params.json'. Note that this file will be stored in the self.ler_directory.
+        save_batch : `bool`
+            if True, the function will save the parameters in batches. if False, the function will save all the parameters at the end of sampling. save_batch=False is faster.
+        resume : `bool`
+            resume = False (default) or True.
+            if True, it appends the new samples to the existing json file.
 
         Returns
         ----------
         lensed_param : `dict`
-            dictionary of lensed GW source parameters.
-            lensed_param.keys() = 
+            dictionary of lensed GW source parameters. Refer to :attr:`~lensed_param` for details.
         """
 
         print("sampling lensed params...")
@@ -1046,6 +1391,7 @@ class LeR(LensGalaxyParameterDistribution):
             if len(lensed_param) == 0:  # if empty
                     lensed_param = lensed_param_
             else:
+                # below will not be used in the first iteration
                 # replace the values of the keys
                 # idx defines the position that does not satisfy the strong lensing condition
                 for key, value in lensed_param_.items():
@@ -1096,15 +1442,21 @@ class LeR(LensGalaxyParameterDistribution):
         self,
         lensed_param=None,
         snr_threshold=[8.0,8.0],
+        pdet_threshold=0.5,
         num_img=[1,1],
         output_jsonfile=None,
         nan_to_num=True,
         detectability_condition="step_function",
         snr_recalculation=False,
-        threshold_snr_recalculation=[6.0,6.0],
+        snr_threshold_recalculation=[[4,4], [20,20]],
     ):
         """
-        Function to calculate the lensed rate. This function also stores the parameters of the detectable events in json file.
+        Function to calculate the lensed rate. This function also stores the parameters of the detectable events in json file. There are two conditions for detectability: 'step_function' and 'pdet'.
+
+        1. 'step_function': If two images have SNR>8.0, then the event is detectable. This is a step function. This is with the assumption that SNR function is provided and not None. 
+        2. 'pdet': 
+            i) If self.pdet is None and self.snr is not None, then it will calculate the pdet from the snr. There is no hard cut for this pdet and can have value ranging from 0 to 1 near the threshold.
+            ii) If self.pdet is not None, then it will use the generated pdet.
 
         Parameters
         ----------
@@ -1112,11 +1464,14 @@ class LeR(LensGalaxyParameterDistribution):
             dictionary of GW source parameters or json file name.
             default lensed_param = 'lensed_params.json'.
         snr_threshold : `float`
-            threshold for detection signal to noise ratio.
+            threshold for detection signal to noise ratio. This is use when self.snr is provided.
             default snr_threshold = [8.0,8.0].
+        pdet_threshold : `float`
+            threshold for detection probability. This is use when self.pdet is provided.
+            default pdet_threshold = 0.5.
         num_img : `int`
-            number of images.
-            default num_img = [1,1].
+            number of images corresponding to the snr_threshold.
+            default num_img = [1,1]. Together with snr_threshold = [8.0,8.0], it means that two images with snr>8.0. Same condition can also be represented by snr_threshold = 8.0 and num_img = 2.
         output_jsonfile : `str`
             json file name for storing the parameters of the detectable events.
             default output_jsonfile = 'lensed_params_detectable.json'.
@@ -1128,18 +1483,18 @@ class LeR(LensGalaxyParameterDistribution):
             default detectability_condition = 'step_function'.
             other options are 'pdet'.
         snr_recalculation : `bool`
-            if True, the SNR of centain events (snr>threshold_snr_recalculation)will be recalculate with 'inner product'. This is useful when the snr is calculated with 'ann' method.
+            if True, the SNR of centain events (snr>snr_threshold_recalculation)will be recalculate with 'inner-product' method. This is useful when the snr is calculated with 'ann' method of `gwsnr`.
             default snr_recalculation = False.
-        threshold_snr_recalculation : `float`
-            threshold for recalculation of detection signal to noise ratio.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+            default snr_threshold_recalculation = [[4,4], [20,20]].
 
         Returns
         ----------
         total_rate : `float`
             total lensed rate (Mpc^-3 yr^-1).
         lensed_param : `dict`
-            dictionary of lensed GW source parameters of the detectable events.
-            lensed_param.keys() =
+            dictionary of lensed GW source parameters of the detectable events. Refer to :attr:`~lensed_param` for details.
 
         Examples
         ----------
@@ -1148,200 +1503,200 @@ class LeR(LensGalaxyParameterDistribution):
         >>> ler.lensed_cbc_statistics();
         >>> total_rate, lensed_param_detectable = ler.lensed_rate()
         """
-        
-        # call self.json_file_names["ler_param"] and for adding the final results
-        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
 
-        # get gw params from json file if not provided
-        if lensed_param is None:
-            lensed_param = self.json_file_names["lensed_param"]
-        if type(lensed_param) == str:
-            self.json_file_names["lensed_param"] = lensed_param
-            path_ = self.ler_directory+"/"+lensed_param
-            print(f"getting lensed_params from json file {path_}...")
-            lensed_param = get_param_from_json(self.ler_directory+"/"+lensed_param)
-        else:
-            print("using provided lensed_param dict...")
-            lensed_param = lensed_param.copy()
+        # load lensed parameters
+        lensed_param = self._load_param(lensed_param, param_type="lensed")
 
-        # check for images with snr above threshold
-        # convert to array
-        snr_threshold_ = np.array([snr_threshold]).reshape(-1)  
-        num_img_ = np.array([num_img]).reshape(-1)
-        # get descending sorted idx of snr_threshold
-        idx = np.argsort(-snr_threshold_)
-        snr_threshold = snr_threshold_[idx]
-        num_img = num_img_[idx]
+        # re-analyse the provided snr_threshold and num_img
+        snr_threshold, num_img = self._check_snr_threshold_lensed(snr_threshold, num_img)
 
         # get size of the lensed_param for a parameter
         total_events = len(lensed_param["zs"])
-        # recalculate snr if required
+
         # this ensures that the snr is recalculated for the detectable events
         # with inner product
+        # below is use when the snr is calculated with 'ann' method of `gwsnr`
         if snr_recalculation:
-            # dealing with provided threshold_snr_recalculation
-            threshold_snr_recalculation = np.array([threshold_snr_recalculation]).reshape(-1)
-            idx = np.argsort(-threshold_snr_recalculation)
-            threshold_snr_recalculation = threshold_snr_recalculation[idx]
-            num_img_recalculation = num_img[idx]
-
-            # check optimal_snr_net is provided in dict
-            snr_param = lensed_param["optimal_snr_net"]
-            if "optimal_snr_net" in lensed_param.keys():
-                snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
-            else:
-                print("optimal_snr_net not provided in lensed_param dict. Exiting...")
-                return None
-            # for each row: choose a threshold and check if the number of images above threshold. Sum over the images. If sum is greater than num_img, then snr_hit = True 
-            j = 0
-            idx_max = 0
-            snr_hit = np.full(total_events, True)  # boolean array to store the result of the threshold condition
-            for i in range(len(threshold_snr_recalculation)):
-                idx_max = idx_max + num_img_recalculation[i]
-                snr_hit = snr_hit & (np.sum((snr_param[:,j:idx_max] > threshold_snr_recalculation[i]), axis=1) >= num_img_recalculation[i])
-                j = idx_max
-
-            # reduce the size of the dict
-            for key, value in lensed_param.items():
-                lensed_param[key] = value[snr_hit]
-            # recalculate more accurate snrs
-            print("calculating snrs...")
-            snrs, lensed_param = self.get_lensed_snrs(
-                snr_calculator=self.snr_bilby,
-                list_of_detectors=self.list_of_detectors,
-                lensed_param=lensed_param,
-            )
-            lensed_param.update(snrs)
-
-        if self.snr:
-            if "optimal_snr_net" not in lensed_param:
-                raise ValueError("'optimal_snr_net' not in lensed parm dict provided")
-            if detectability_condition == "step_function":
-                print("given detectability_condition == 'step_function'")
-                
-                if "optimal_snr_net" in lensed_param.keys():
-                    snr_param = lensed_param["optimal_snr_net"]
-                    snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
-                else:
-                    print("snr not provided in lensed_param dict. Exiting...")
-                    return None
-                snr_hit = np.full(len(snr_param), True)  # boolean array to store the result of the threshold condition
-                
-                # for each row: choose a threshold and check if the number of images above threshold. Sum over the images. If sum is greater than num_img, then snr_hit = True 
-                j = 0
-                idx_max = 0
-                for i in range(len(snr_threshold)):
-                    idx_max = idx_max + num_img[i]
-                    snr_hit = snr_hit & (np.sum((snr_param[:,j:idx_max] > snr_threshold[i]), axis=1) >= num_img[i])
-                    j = idx_max
-
-                # montecarlo integration
-                total_rate = self.normalization_pdf_z_lensed * np.sum(snr_hit)/total_events
-                print("total lensed rate (yr^-1) (with step function): {}".format(total_rate))
-            
-            elif detectability_condition == "pdet":
-                print("given detectability_condition == 'pdet'")
-                # pdet dimension is (size, n_max_images)
-                snr_param = lensed_param["optimal_snr_net"]
-                snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
-
-                pdet = np.ones(np.shape(snr_param))
-                j = 0
-                idx_max = 0
-                for i in range(len(snr_threshold)):
-                    idx_max = idx_max + num_img[i]
-                    pdet[:,j:idx_max] = 1 - norm.cdf(snr_threshold[i] - snr_param[:,j:idx_max])
-                    j = idx_max
-
-                snr_hit = np.prod(pdet, axis=1)>0.5
-                # montecarlo integration
-                total_rate = self.normalization_pdf_z_lensed * np.sum(snr_hit)/total_events
-                print(f"total lensed rate (yr^-1) (with pdet function): {total_rate}")
-
-        elif self.pdet:
-            # check if pdet is provided in lensed_param dict
-            if "pdet_net" in lensed_param.keys():
-                print("given detectability_condition == 'pdet'")
-                pdet = lensed_param["pdet_net"]
-                pdet = -np.sort(-pdet, axis=1)  # sort pdet in descending order
-                pdet = pdet[:,:np.sum(num_img)]  # use only num_img images
+            lensed_param = self._recalculate_snr_lensed(lensed_param, snr_threshold_recalculation, num_img, total_events)
 
-                snr_hit = np.prod(pdet, axis=1)>0.5
-                # montecarlo integration
-                total_rate = self.normalization_pdf_z_lensed * np.sum(snr_hit)/total_events
-                print(f"total lensed rate (yr^-1) (with pdet function): {total_rate}")
+        snr_hit = self._find_detectable_index_lensed(lensed_param, snr_threshold, pdet_threshold, num_img, detectability_condition)
 
-            else:
-                raise ValueError("'pdet_net' not in unlensed parm dict provided")
+        # montecarlo integration
+        total_rate = self.rate_function(np.sum(snr_hit), total_events, param_type="lensed")
 
-        print(f"number of simulated lensed detectable events: {np.sum(snr_hit)}")
-        print(f"number of simulated all lensed events: {total_events}")
-        # store all detectable params in json file
-        if nan_to_num:
-            for key, value in lensed_param.items():
-                lensed_param[key] = np.nan_to_num(value[snr_hit])
-        else:
-            for key, value in lensed_param.items():
-                lensed_param[key] = value[snr_hit]
-            
         # store all detectable params in json file
-        if output_jsonfile is None:
-            output_jsonfile = self.json_file_names["lensed_param_detectable"]
-        else:
-            self.json_file_names["lensed_param_detectable"] = output_jsonfile
-        path_ = self.ler_directory+"/"+output_jsonfile
-        print(f"storing detectable lensed params in {path_}")
-        append_json(self.ler_directory+"/"+output_jsonfile, lensed_param, replace=True)
+        self._save_detectable_params(output_jsonfile, lensed_param, snr_hit, key_file_name="lensed_param_detectable", nan_to_num=nan_to_num, verbose=True, replace_jsonfile=True)
 
-        # write the results
-        data["detectable_lensed_rate_per_year"] = total_rate
-        data["detectability_condition"] = detectability_condition
-        append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
+        # append ler_param and save it
+        self._append_ler_param(total_rate, detectability_condition, param_type="lensed")
 
         return total_rate, lensed_param
-     
-    def rate_ratio(self):
+
+    def _check_snr_threshold_lensed(self, snr_threshold, num_img):
         """
-        Function to calculate and display unlensed and lensed merger rate ratio. 
-        It will get the unlensed_rate and lensed_rate from self.json_file_ler_param
+        Helper function to check the snr_threshold and num_img for lensed events.
 
-        Returns
+        Parameters
         ----------
-        rate_ratio : `float`
-            rate ratio.
+        snr_threshold : `float`
+            threshold for detection signal to noise ratio.
+            default snr_threshold = [8.0,8.0].
+        num_img : `int`
+            number of images corresponding to the snr_threshold.
+            default num_img = [1,1]. Together with snr_threshold = [8.0,8.0], it means that two images with snr>8.0. Same condition can also be represented by snr_threshold = 8.0 and num_img = 2.
 
-        Examples
+        Returns
         ----------
-        >>> from ler.rates import LeR
-        >>> ler = LeR()
-        >>> ler.unlensed_cbc_statistics();
-        >>> ler.lensed_cbc_statistics();
-        >>> ler.unlensed_rate();
-        >>> ler.lensed_rate();
-        >>> ler.rate_ratio()
+        snr_threshold : `float`
+            threshold for detection signal to noise ratio.
+        num_img : `int`
+            number of images corresponding to the snr_threshold.
         """
+        # check for images with snr above threshold
+        # convert to array
+        snr_threshold_ = np.array([snr_threshold]).reshape(-1)  
+        num_img_ = np.array([num_img]).reshape(-1)
+        # get descending sorted idx of snr_threshold
+        idx = np.argsort(-snr_threshold_)
+        snr_threshold = snr_threshold_[idx]
+        num_img = num_img_[idx]
 
-        # call json_file_ler_param and add the results
-        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
+        return snr_threshold, num_img
 
-        try:
-            unlensed_rate = data['detectable_unlensed_rate_per_year']
-            lensed_rate = data['detectable_lensed_rate_per_year']
-        except:
-            raise ValueError(f"detectable_unlensed_rate_per_year or 'detectable_lensed_rate_per_year' not found in {self.json_file_names['ler_params']} json file. Exiting...")
-        
-        rate_ratio = unlensed_rate / lensed_rate
-        # append the results
-        data['rate_ratio'] = rate_ratio
-        # write the results
-        append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
-        
-        print(f"unlensed_rate: {unlensed_rate}")
-        print(f"lensed_rate: {lensed_rate}")
-        print(f"ratio: {rate_ratio}")
+    def _recalculate_snr_lensed(self, lensed_param, snr_threshold_recalculation, num_img, total_events):
+        """
+        Helper function to recalculate the SNR of lensed events with SNR above a given threshold.
+
+        Parameters
+        ----------
+        lensed_param : `dict`
+            dictionary of lensed GW source parameters.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+        num_img : `int`
+            number of images corresponding to the snr_threshold.
+        total_events : `int`
+            total number of events.
+
+        Returns
+        ----------
+        lensed_param : `dict`
+            dictionary of lensed GW source parameters.
+        """
+
+        # dealing with provided snr_threshold_recalculation
+        snr_threshold_recalculation_min, num_img_recalculation = self._check_snr_threshold_lensed(snr_threshold_recalculation[0], num_img)
+        snr_threshold_recalculation_max, _ = self._check_snr_threshold_lensed(snr_threshold_recalculation[1], num_img)
+
+        # check optimal_snr_net is provided in dict
+        if "optimal_snr_net" not in lensed_param:
+            raise ValueError("optimal_snr_net not provided in lensed_param dict. Exiting...")
+
+        snr_param = lensed_param["optimal_snr_net"]
+        snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
+            
+        # for each row: choose a threshold and check if the number of images above threshold. Sum over the images. If sum is greater than num_img, then snr_hit = True
+        j = 0
+        idx_max = 0
+        snr_hit = np.full(total_events, True)  # boolean array to store the result of the threshold condition
+        for i, snr_th_min in enumerate(snr_threshold_recalculation_min):
+            snr_th_max = snr_threshold_recalculation_max[i]
+            idx_max = idx_max + num_img_recalculation[i]
+            snr_hit = snr_hit & (np.sum((snr_param[:,j:idx_max] > snr_th_min) & (snr_param[:,j:idx_max] < snr_th_max), axis=1) >= num_img_recalculation[i])
+            j = idx_max
+
+        # reduce the size of the dict
+        for key, value in lensed_param.items():
+            lensed_param[key] = value[snr_hit]
+        # recalculate more accurate snrs
+        print("calculating snrs...")
+        snrs, lensed_param = self.get_lensed_snrs(
+            snr_calculator=self.snr_bilby,
+            list_of_detectors=self.list_of_detectors,
+            lensed_param=lensed_param,
+        )
+        lensed_param.update(snrs)
+
+        return lensed_param
+
+    def _find_detectable_index_lensed(self, lensed_param, snr_threshold, pdet_threshold, num_img, detectability_condition):
+        """
+        Helper function to find the index of detectable events based on SNR or p_det.
+
+        Parameters
+        ----------
+        lensed_param : `dict`
+            dictionary of lensed GW source parameters.
+        snr_threshold : `float` or `list`
+            threshold for detection signal to noise ratio.
+            default snr_threshold = [8.0,8.0].
+        pdet_threshold : `float` or `list`
+            threshold for detection probability.
+            default pdet_threshold = 0.5.
+        num_img : `int` or `list`
+            number of images corresponding to the snr_threshold.
+            default num_img = [1,1].
+        detectability_condition : `str`
+            detectability condition.
+            default detectability_condition = 'step_function'.
+            other options are 'pdet'.
+
+        Returns
+        ----------
+        snr_hit : `bool`
+            boolean array to store the result of the threshold condition.
+        """
+
+        print(f"given detectability_condition == {detectability_condition}")
+        if detectability_condition == "step_function":
+            if "optimal_snr_net" not in lensed_param:
+                raise ValueError("'optimal_snr_net' not in lensed parm dict provided")
+            snr_param = lensed_param["optimal_snr_net"]
+            snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
+            snr_hit = np.full(len(snr_param), True)  # boolean array to store the result of the threshold condition
+
+            # for each row: choose a threshold and check if the number of images above threshold. Sum over the images. If sum is greater than num_img, then snr_hit = True 
+            # algorithm: 
+            # i) consider snr_threshold=[8,6] and num_img=[2,1] and first row of snr_param[0]=[12,8,6,1]. Note that the snr_param is sorted in descending order.
+            # ii) for loop runs wrt snr_threshold. idx_max = idx_max + num_img[i]
+            # iii) First iteration: snr_threshold=8 and num_img=2. In snr_param, column index 0 and 1 (i.e. 0:num_img[0]) are considered. The sum of snr_param[0, 0:2] > 8 is checked. If True, then snr_hit = True. 
+            # v) Second iteration: snr_threshold=6 and num_img=1. In snr_param, column index 2 (i.e. num_img[0]:num_img[1]) is considered. The sum of snr_param[0, 0:1] > 6 is checked. If True, then snr_hit = True.
+            j = 0
+            idx_max = 0
+            for i, snr_th in enumerate(snr_threshold):
+                idx_max = idx_max + num_img[i]
+                snr_hit = snr_hit & (np.sum((snr_param[:,j:idx_max] > snr_th), axis=1) >= num_img[i])
+                j = idx_max
+                
+        elif detectability_condition == "pdet":
+            if "pdet_net" not in lensed_param:
+                if "optimal_snr_net" not in lensed_param:
+                    raise ValueError("'optimal_snr_net' or 'pdet_net' not in lensed parm dict provided")
+                else:
+                    print("calculating pdet using 'optimal_snr_net'...")
+                    # pdet dimension is (size, n_max_images)
+                    snr_param = lensed_param["optimal_snr_net"]
+                    snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
+
+                    # column index beyong np.sum(num_img)-1 are not considered
+                    pdet = np.ones((len(snr_param), np.sum(num_img)))
+                    j = 0
+                    idx_max = 0
+                    for i, snr_th in enumerate(snr_threshold):
+                        idx_max = idx_max + num_img[i]
+                        pdet[:,j:idx_max] = 1 - norm.cdf(snr_th - snr_param[:,j:idx_max])
+                        j = idx_max
+            else:
+                pdet = lensed_param["pdet_net"]
+                # sort pdet in descending order
+                pdet = -np.sort(-pdet, axis=1)  
+                # column index beyong np.sum(num_img)-1 are not considered
+                pdet = pdet[:,:np.sum(num_img)] 
+                
+            snr_hit = np.prod(pdet, axis=1)>=pdet_threshold
+
+        return snr_hit
 
-        return unlensed_rate / lensed_rate
-    
     def rate_comparision_with_rate_calculation(
         self,
         unlensed_param=None,
@@ -1355,7 +1710,7 @@ class LeR(LensGalaxyParameterDistribution):
         detectability_condition="step_function",
     ):
         """
-        Function to calculate the unlensed and lensed rate and compare by computing the ratio. This function also stores the parameters of the detectable events in json file.
+        Function to calculate the unlensed and lensed rate and compare by computing the ratio. This function also stores the parameters of the detectable events in json file. If you use this function, you do not need to call the functions unlensed_rate and lensed_rate separately.
 
         Parameters
         ----------
@@ -1364,21 +1719,27 @@ class LeR(LensGalaxyParameterDistribution):
             default unlensed_param = 'unlensed_params.json'.
         snr_threshold_unlensed : `float`
             threshold for detection signal to noise ratio.
-            e.g. snr_threshold = 8.
+            e.g. snr_threshold_unlensed = 8.
         output_jsonfile_unlensed : `str`
             json file name for storing the parameters of the detectable events.
-            default output_jsonfile = 'unlensed_params_detectable.json'.
+            default output_jsonfile_unlensed = 'unlensed_params_detectable.json'.
         lensed_param : `dict` or `str`
             dictionary of GW source parameters or json file name.
             default lensed_param = 'lensed_params.json'.
         snr_threshold_lensed : `float`
             threshold for detection signal to noise ratio.
-            e.g. snr_threshold = 8.
+            default snr_threshold_lensed = [8.0,8.0].
+        num_img : `int`
+            number of images.
+            default num_img = [1,1]. Together with snr_threshold = [8.0,8.0], it means that two images with snr>8.0. Same condition can also be represented by snr_threshold = 8.0 and num_img = 2.
         output_jsonfile_lensed : `str`
             json file name for storing the parameters of the detectable events.
-            default output_jsonfile = 'lensed_params_detectable.json'.
+            default output_jsonfile_lensed = 'lensed_params_detectable.json'.
+        nan_to_num : `bool`
+            if True, nan values will be converted to 0.
+            default nan_to_num = True.
         detectability_condition : `str`
-            detectability condition. 
+            detectability condition.
             default detectability_condition = 'step_function'.
             other options are 'pdet'.
 
@@ -1387,9 +1748,9 @@ class LeR(LensGalaxyParameterDistribution):
         rate_ratio : `float`
             rate ratio.
         unlensed_param : `dict`
-            dictionary of unlensed GW source parameters of the detectable events.
+            dictionary of unlensed GW source parameters of the detectable events. Refer to :attr:`~unlensed_param` for details.
         lensed_param : `dict`
-            dictionary of lensed GW source parameters of the detectable events.
+            dictionary of lensed GW source parameters of the detectable events. Refer to :attr:`~lensed_param` for details.
 
         Examples
         ----------
@@ -1420,136 +1781,168 @@ class LeR(LensGalaxyParameterDistribution):
             detectability_condition=detectability_condition,
         )
         # calculate rate ratio
+        rate_ratio = self.rate_ratio()
+
+        return rate_ratio, unlensed_param, lensed_param
+     
+    def rate_ratio(self):
+        """
+        Function to calculate and display unlensed and lensed merger rate ratio. It will get the unlensed_rate and lensed_rate from files corresponding to the names included in self.json_file_ler_param. 
+
+        Returns
+        ----------
+        rate_ratio : `float`
+            rate ratio.
+
+        Examples
+        ----------
+        >>> from ler.rates import LeR
+        >>> ler = LeR()
+        >>> ler.unlensed_cbc_statistics();
+        >>> ler.lensed_cbc_statistics();
+        >>> ler.unlensed_rate();
+        >>> ler.lensed_rate();
+        >>> ler.rate_ratio()
+        """
+
+        # call json_file_ler_param and add the results
+        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
+
+        try:
+            unlensed_rate = data['detectable_unlensed_rate_per_year']
+            lensed_rate = data['detectable_lensed_rate_per_year']
+        except:
+            raise ValueError(f"detectable_unlensed_rate_per_year or 'detectable_lensed_rate_per_year' not found in {self.json_file_names['ler_params']} json file. Exiting...")
+        
         rate_ratio = unlensed_rate / lensed_rate
         # append the results
-        data['detectable_unlensed_rate_per_year'] = unlensed_rate
-        data['detectable_lensed_rate_per_year'] = lensed_rate
         data['rate_ratio'] = rate_ratio
-        data['detectability_condition'] = detectability_condition
         # write the results
         append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
         
-        print(f"unlensed_rate (per year): {unlensed_rate}")
-        print(f"lensed_rate (per year): {lensed_rate}")
+        print(f"unlensed_rate: {unlensed_rate}")
+        print(f"lensed_rate: {lensed_rate}")
         print(f"ratio: {rate_ratio}")
 
-        return rate_ratio, unlensed_param, lensed_param
+        return unlensed_rate / lensed_rate
 
     def selecting_n_unlensed_detectable_events(
         self,
         size=100,
         batch_size=None,
         snr_threshold=8.0,
+        pdet_threshold=0.5,
         resume=False,
         output_jsonfile="n_unlensed_param_detectable.json",
         meta_data_file="meta_unlensed.json",
+        detectability_condition="step_function",
         trim_to_size=True,
+        snr_recalculation=False,
+        snr_threshold_recalculation=[4, 12],
     ):
         """
-        Function to select n unlensed detectable events.
+        Function to generate n unlensed detectable events. This fuction samples the unlensed parameters and save only the detectable events in json file. It also records metadata in the JSON file, which includes the total number of events and the cumulative rate of events. This functionality is particularly useful for generating a fixed or large number of detectable events until the event rates stabilize.
 
         Parameters
         ----------
         size : `int`
             number of samples to be selected.
             default size = 100.
+        batch_size : `int`
+            batch size for sampling.
+            default batch_size = 50000.
         snr_threshold : `float`
             threshold for detection signal to noise ratio.
             e.g. snr_threshold = 8.
+        pdet_threshold : `float`
+            threshold for detection probability.
+            default pdet_threshold = 0.5.
         resume : `bool`
-            if True, it will resume the sampling from the last batch.
-            default resume = False.
+            resume = False (default) or True.
+            if True, the function will resume from the last batch.
         output_jsonfile : `str`
-            json file name for storing the parameters.
-            default output_jsonfile = 'n_unlensed_params_detectable.json'.
+            json file name for storing the parameters of the detectable events.
+            default output_jsonfile = 'n_unlensed_param_detectable.json'.
+        meta_data_file : `str`
+            json file name for storing the metadata.
+            default meta_data_file = 'meta_unlensed.json'.
+        detectability_condition : `str`
+            detectability condition.
+            default detectability_condition = 'step_function'.
+            other options are 'pdet'.
+        trim_to_size : `bool`
+            if True, the final result will be trimmed to size.
+            default trim_to_size = True.
+        snr_recalculation : `bool`
+            if True, the SNR of centain events (snr>snr_threshold_recalculation)will be recalculate with 'inner-product' method. This is useful when the snr is calculated with 'ann' method of `gwsnr`.
+            default snr_recalculation = False.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+            default snr_threshold_recalculation = [4, 12].
 
         Returns
         ----------
         param_final : `dict`
-            dictionary of unlensed GW source parameters of the detectable events.
-            param_final.keys() = ['zs', 'geocent_time', 'ra', 'dec', 'phase', 'psi', 'theta_jn', 'luminosity_distance', 'mass_1_source', 'mass_2_source', 'mass_1', 'mass_2', 'optimal_snr_net', 'L1', 'H1', 'V1']
+            dictionary of unlensed GW source parameters of the detectable events. Refer to :attr:`~unlensed_param` for details.
 
         Examples
         ----------
         >>> from ler.rates import LeR
         >>> ler = LeR()
-        >>> unlensed_param_final = ler.selecting_n_unlensed_detectable_events(size=500)
+        >>> unlensed_param = ler.selecting_n_unlensed_detectable_events(size=100)
         """
 
-        meta_data_path = self.ler_directory+"/"+meta_data_file
-        output_path = self.ler_directory+"/"+output_jsonfile
-            
-        if batch_size is None:
-            batch_size = self.batch_size
-        else:
-            self.batch_size = batch_size
+        # initial setup
+        n, events_total, output_path, meta_data_path, buffer_file = self._initial_setup_for_n_event_selection(meta_data_file, output_jsonfile, resume, batch_size)
 
-        if not resume:
-            n = 0  # iterator
-            events_total = 0
-            if os.path.exists(output_path):
-                os.remove(output_path)
-            if os.path.exists(meta_data_path):
-                os.remove(meta_data_path)
-        else:
-            # get sample size as size from json file
-            if os.path.exists(output_path):
-                param_final = get_param_from_json(output_path)
-                n = len(param_final["zs"])
-                events_total = load_json(meta_data_path)["events_total"][-1]
-                del param_final
-            else:
-                n = 0
-                events_total = 0
-
-        buffer_file = self.ler_directory+"/"+"unlensed_params_buffer.json"
-        print("collected number of detectable events = ", n)
-        
+        # loop until n samples are collected
         while n < size:
             # disable print statements
             with contextlib.redirect_stdout(None):
-                self.dict_buffer = None
+                self.dict_buffer = None  # this is used to store the sampled unlensed_param in batches when running the sampling_routine
                 unlensed_param = self.unlensed_sampling_routine(
                     size=batch_size, output_jsonfile=buffer_file, save_batch=False,resume=False
                 )
 
-            # get snr
-            snr = unlensed_param["optimal_snr_net"]
-            # index of detectable events
-            idx = snr > snr_threshold
+            total_events_in_this_iteration = len(unlensed_param["zs"])
+            # below is use when the snr is calculated with 'ann' method of `gwsnr`
+            if snr_recalculation:
+                # select only above centain snr threshold
+                unlensed_param = self._recalculate_snr_unlensed(unlensed_param, snr_threshold_recalculation)
 
-            # store all params in json file
-            for key, value in unlensed_param.items():
-                unlensed_param[key] = value[idx]
-            append_json(file_name=output_path, new_dictionary=unlensed_param, replace=False)
+            # find index of detectable events
+            idx_detectable = self._find_detectable_index_unlensed(unlensed_param, snr_threshold, pdet_threshold, detectability_condition)
 
-            n += np.sum(idx)
-            events_total += len(idx)                
-            total_rate = self.normalization_pdf_z * n / events_total
+            # store all params in json file
+            self._save_detectable_params(output_jsonfile, unlensed_param, idx_detectable, key_file_name="n_unlensed_detectable_events", nan_to_num=False, verbose=False, replace_jsonfile=False)
 
-            # save meta data
-            meta_data = dict(events_total=[events_total], detectable_events=[float(n)], total_rate=[total_rate])
-            if os.path.exists(meta_data_path):
-                append_json(file_name=meta_data_path, new_dictionary=meta_data, replace=False)
-            else:
-                append_json(file_name=meta_data_path, new_dictionary=meta_data, replace=True)
+            n += np.sum(idx_detectable)
+            events_total += total_events_in_this_iteration              
+            total_rate = self.rate_function(n, events_total, param_type="unlensed", verbose=False)
 
-            print("collected number of detectable events = ", n)
-            print("total number of events = ", events_total)
-            print(f"total unlensed rate (yr^-1): {total_rate}")
+            # bookmark
+            self._append_meta_data(meta_data_path, n, events_total, total_rate)
 
-        print(f"storing detectable unlensed params in {output_path}")
+        print(f"stored detectable unlensed params in {output_path}")
+        print(f"stored meta data in {meta_data_path}")
 
         if trim_to_size:
-            # trim the final param dictionary
-            print(f"\n trmming final result to size={size}")
+            param_final, total_rate = self._trim_results_to_size(size, output_path, meta_data_path)
+        else:
             param_final = get_param_from_json(output_path)
-            # trim the final param dictionary, randomly, without repeating
-            for key, value in param_final.items():
-                param_final[key] = param_final[key][:size]
 
-        # save the final param dictionary
-        append_json(output_path, param_final, replace=True)
+        # call self.json_file_names["ler_param"] and for adding the final results
+        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
+        # write the results
+        try:
+            data["detectable_unlensed_rate_per_year"] = total_rate
+            data["detectability_condition_unlensed"] = detectability_condition
+        except:
+            meta = get_param_from_json(meta_data_path)
+            data["detectable_unlensed_rate_per_year"] = meta["total_rate"][-1]
+            data["detectability_condition_unlensed"] = detectability_condition
+
+        append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
 
         return param_final
     
@@ -1557,55 +1950,166 @@ class LeR(LensGalaxyParameterDistribution):
         self,
         size=100,
         batch_size=None,
-        snr_threshold=8.0,
-        num_img=2,
+        snr_threshold=[8.0,8.0],
+        pdet_threshold=0.5,
+        num_img=[1,1],
         resume=False,
         detectability_condition="step_function",
         output_jsonfile="n_lensed_params_detectable.json",
         meta_data_file="meta_lensed.json",
         trim_to_size=True,
         nan_to_num=False,
+        snr_recalculation=False,
+        snr_threshold_recalculation=[[4,4],[12,12]],
     ):
         """
-        Function to select n lensed detectable events.
+        Function to generate n lensed detectable events. This fuction only samples the lensed parameters and save only the detectable events in json file. It also records metadata in the JSON file, which includes the total number of events and the cumulative rate of events. This functionality is particularly useful for generating a fixed or large number of detectable events until the event rates stabilize.
 
         Parameters
         ----------
         size : `int`
-            number of samples to be selected.
+            number of samples.
             default size = 100.
+        batch_size : `int`
+            batch size for sampling.
+            default batch_size = 50000.
         snr_threshold : `float`
             threshold for detection signal to noise ratio.
-            e.g. snr_threshold = 8.
+            default snr_threshold = [8.0,8.0].
+        pdet_threshold : `float`
+            threshold for detection probability.
+            default pdet_threshold = 0.5.
         num_img : `int`
             number of images.
-            default num_img = 2.
+            default num_img = [1,1]. Together with snr_threshold = [8.0,8.0], it means that two images with snr>8.0. Same condition can also be represented by snr_threshold = 8.0 and num_img = 2.
         resume : `bool`
-            if True, it will resume the sampling from the last batch.
-            default resume = False.
+            resume = False (default) or True.
+            if True, it appends the new samples to the existing json file.
         detectability_condition : `str`
             detectability condition.
             default detectability_condition = 'step_function'.
             other options are 'pdet'.
         output_jsonfile : `str`
-            json file name for storing the parameters.
+            json file name for storing the parameters of the detectable events.
+            default output_jsonfile = 'n_lensed_params_detectable.json'.
+        meta_data_file : `str`
+            json file name for storing the metadata.
+            default meta_data_file = 'meta_lensed.json'.
+        trim_to_size : `bool`
+            if True, the final result will be trimmed to size.
+            default trim_to_size = True.
+        nan_to_num : `bool`
+            if True, nan values will be converted to 0.
+            default nan_to_num = False.
+        snr_recalculation : `bool`
+            if True, the SNR of centain events (snr>snr_threshold_recalculation)will be recalculate with 'inner-product' method. This is useful when the snr is calculated with 'ann' method of `gwsnr`.    
+            default snr_recalculation = False.
+        snr_threshold_recalculation : `list`
+            lower and upper threshold for recalculation of detection signal to noise ratio.
+            default snr_threshold_recalculation = [[4,4], [12,12]].
 
         Returns
         ----------
         param_final : `dict`
-            dictionary of lensed GW source parameters of the detectable events.
-            param_final.keys() =
+            dictionary of lensed GW source parameters of the detectable events. Refer to :attr:`~lensed_param` for details.
 
         Examples
         ----------
         >>> from ler.rates import LeR
         >>> ler = LeR()
-        >>> lensed_param_final = ler.selecting_n_lensed_detectable_events(size=500)
+        >>> lensed_param = ler.selecting_n_lensed_detectable_events(size=100)
+        """
+
+        # initial setup
+        n, events_total, output_path, meta_data_path, buffer_file = self._initial_setup_for_n_event_selection(meta_data_file, output_jsonfile, resume, batch_size)
+
+        # re-analyse the provided snr_threshold and num_img
+        snr_threshold, num_img = self._check_snr_threshold_lensed(snr_threshold, num_img)
+
+        while n < size:
+            # disable print statements
+            with contextlib.redirect_stdout(None):
+                self.dict_buffer = None  # this is used to store the sampled lensed_param in batches when running the sampling_routine
+                lensed_param = self.lensed_sampling_routine(
+                    size=self.batch_size, output_jsonfile=buffer_file, resume=False
+                )  # Dimensions are (size, n_max_images)
+
+            total_events_in_this_iteration = len(lensed_param["zs"])
+
+            # Below code ensures that the snr is recalculated for the detectable events with inner product
+            # The code is use when the snr is calculated with 'ann' method of `gwsnr`
+            if snr_recalculation:
+                lensed_param = self._recalculate_snr_lensed(lensed_param, snr_threshold_recalculation, num_img, total_events_in_this_iteration)
+
+            snr_hit = self._find_detectable_index_lensed(lensed_param, snr_threshold, pdet_threshold, num_img, detectability_condition)
+                    
+            # store all params in json file
+            self._save_detectable_params(output_jsonfile, lensed_param, snr_hit, key_file_name="n_lensed_detectable_events", nan_to_num=nan_to_num, verbose=False, replace_jsonfile=False)
+
+            n += np.sum(snr_hit)
+            events_total += total_events_in_this_iteration
+            total_rate = self.rate_function(n, events_total, param_type="lensed", verbose=False)
+
+            # save meta data
+            self._append_meta_data(meta_data_path, n, events_total, total_rate)
+
+        print(f"storing detectable lensed params in {output_path}")
+        print(f"storing meta data in {meta_data_path}")
+
+        if trim_to_size:
+            param_final, total_rate = self._trim_results_to_size(size, output_path, meta_data_path, param_type="lensed")
+        else:
+            param_final = get_param_from_json(output_path)
+
+        # call self.json_file_names["ler_param"] and for adding the final results
+        data = load_json(self.ler_directory+"/"+self.json_file_names["ler_params"])
+        # write the results
+        try:
+            data["detectable_lensed_rate_per_year"] = total_rate
+            data["detectability_condition_lensed"] = detectability_condition
+        except:
+            meta = get_param_from_json(meta_data_path)
+            data["detectable_lensed_rate_per_year"] = meta["total_rate"][-1]
+            data["detectability_condition_lensed"] = detectability_condition
+        append_json(self.ler_directory+"/"+self.json_file_names["ler_params"], data, replace=True)
+
+        return param_final
+
+    def _initial_setup_for_n_event_selection(self, meta_data_file, output_jsonfile, resume, batch_size):
+        """Helper function for selecting_n_unlensed_detectable_events and selecting_n_lensed_detectable_events functions. 
+
+        Parameters
+        ----------
+        meta_data_file : `str`
+            json file name for storing the metadata.
+        output_jsonfile : `str`
+            json file name for storing the parameters of the detectable events.
+        resume : `bool`
+            resume = False (default) or True.
+            if True, the function will resume from the last batch.
+        batch_size : `int`
+            batch size for sampling.
+            default batch_size = 50000.
+
+        Returns
+        ----------
+        n : `int`
+            iterator.
+        events_total : `int`
+            total number of events.
+        output_path : `str`
+            path to the output json file.
+        meta_data_path : `str`
+            path to the metadata json file.
+        buffer_file : `str`
+            path to the buffer json file.
         """
 
         meta_data_path = self.ler_directory+"/"+meta_data_file
         output_path = self.ler_directory+"/"+output_jsonfile
-
+        if meta_data_path==output_path:
+            raise ValueError("meta_data_file and output_jsonfile cannot be same.")
+            
         if batch_size is None:
             batch_size = self.batch_size
         else:
@@ -1620,118 +2124,105 @@ class LeR(LensGalaxyParameterDistribution):
                 os.remove(meta_data_path)
         else:
             # get sample size as size from json file
-            try:
+            if os.path.exists(output_path):
                 param_final = get_param_from_json(output_path)
-                n = load_json(meta_data_path)["detectable_events"][-1]
+                n = len(param_final["zs"])
                 events_total = load_json(meta_data_path)["events_total"][-1]
                 del param_final
-            except:
+            else:
                 n = 0
                 events_total = 0
 
-        # check for images with snr above threshold
-        # convert to array
-        snr_threshold = np.array([snr_threshold]).reshape(-1)  
-        num_img = np.array([num_img]).reshape(-1)
-        # get descending sorted idx of snr_threshold
-        idx = np.argsort(-snr_threshold)
-        snr_threshold = snr_threshold[idx]
-        num_img = num_img[idx]
-
-        buffer_file = self.ler_directory+"/"+"lensed_params_buffer.json"
+        buffer_file = "params_buffer.json"
         print("collected number of detectable events = ", n)
-        # loop until n samples are collected
-        while n < size:
-            # disable print statements
-            with contextlib.redirect_stdout(None):
-                self.dict_buffer = None
-                lensed_param = self.lensed_sampling_routine(
-                    size=self.batch_size, output_jsonfile=buffer_file, resume=False
-                )  # Dimensions are (size, n_max_images)
-
-            if detectability_condition == "step_function":
-                snr_hit = np.full(len(lensed_param["zs"]), True)  # boolean array to store the result of the threshold condition
-                try:
-                    snr_param = lensed_param["optimal_snr_net"]
-                    snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
-                except:
-                    raise ValueError("snr not provided in lensed_param dict. Exiting...")
-                
-                # for each row: choose a threshold and check if the number of images above threshold. Sum over the images. If sum is greater than num_img, then snr_hit = True 
-                j = 0
-                idx_max = 0
-                for i in range(len(snr_threshold)):
-                    idx_max = idx_max + num_img[i]
-                    snr_hit = snr_hit & (np.sum((snr_param[:,j:idx_max] > snr_threshold[i]), axis=1) >= num_img[i])
-                    j = idx_max
-
-            elif detectability_condition == "pdet":
-                # check if pdet is provided in unlensed_param dict
-                if "pdet_net" in lensed_param.keys():
-                    pdet = lensed_param["pdet_net"]
-                    pdet = -np.sort(-pdet, axis=1)  # sort pdet in descending order
-                    pdet = pdet[:,:np.sum(num_img)]  # use only num_img images
-                else:
-                    if "optimal_snr_net" in lensed_param.keys():
-                        # pdet dimension is (size, n_max_images)
-                        snr_param = lensed_param["optimal_snr_net"]
-                        snr_param = -np.sort(-snr_param, axis=1)  # sort snr in descending order
-
-                        pdet = np.ones(np.shape(snr_param))
-                        j = 0
-                        idx_max = 0
-                        for i in range(len(snr_threshold)):
-                            idx_max = idx_max + num_img[i]
-                            pdet[:,j:idx_max] = 1 - norm.cdf(snr_threshold[i] - snr_param[:,j:idx_max])
-                            j = idx_max
-                    else:
-                        print("pdet or optimal_snr_net not provided in lensed_param dict. Exiting...")
-                        return None
-                    
-                snr_hit = np.prod(pdet, axis=1)>0.5
-                    
-            # store all params in json file
-            if nan_to_num:
-                for key, value in lensed_param.items():
-                    lensed_param[key] = np.nan_to_num(value[snr_hit])
-            else:
-                for key, value in lensed_param.items():
-                    lensed_param[key] = value[snr_hit]
 
-            if os.path.exists(output_path):
-                append_json(output_path, lensed_param, replace=False)
-            else:
-                append_json(output_path, lensed_param, replace=True)
+        return n, events_total, output_path, meta_data_path, buffer_file
 
-            n += np.sum(snr_hit)
-            events_total += len(snr_hit)
+    def _trim_results_to_size(self, size, output_path, meta_data_path, param_type="unlensed"):
+        """
+        Helper function of 'selecting_n_unlensed_detectable_events' and 'selecting_n_lensed_detectable_events' functions. Trims the data in the output file to the specified size and updates the metadata accordingly.
 
-            # save meta data
-            total_rate = self.normalization_pdf_z_lensed * n / events_total
-            meta_data = dict(events_total=[events_total], detectable_events=[float(n)], total_rate=[total_rate])
-            if os.path.exists(meta_data_path):
-                append_json(meta_data_path, meta_data, replace=False)
-            else:
-                append_json(meta_data_path, meta_data, replace=True)
+        Parameters
+        ----------
+        size : `int`
+            number of samples to be selected.
+        output_path : `str`
+            path to the output json file.
+        meta_data_path : `str`
+            path to the metadata json file.
+        param_type : `str`
+            type of parameters.
+            default param_type = "unlensed".
+            other options are "lensed".
 
-            print("collected number of events = ", n)
-            print("total number of events = ", events_total)
-            print(f"total lensed rate (yr^-1): {total_rate}")
+        Returns
+        ----------
+        param_final : `dict`
+            dictionary of GW source parameters of the detectable events. Refer to :attr:`~unlensed_param` or :attr:`~lensed_param` for details.
+        new_total_rate : `float`
+            total rate (Mpc^-3 yr^-1).
+        """
 
-        print(f"storing detectable lensed params in {output_path}")
+        print(f"\n trmming final result to size={size}")
+        param_final = get_param_from_json(output_path)
+        # randomly select size number of samples
+        len_ = len(list(param_final.values())[0])
+        idx = np.random.choice(len_, size, replace=False)
+        # trim the final param dictionary, randomly, without repeating
+        for key, value in param_final.items():
+            param_final[key] = value[idx]
+
+        # change meta data
+        meta_data = load_json(meta_data_path)
+        old_events_total = meta_data["events_total"][-1]
+        old_detectable_events = meta_data["detectable_events"][-1]
+
+        # adjust the meta data
+        # following is to keep rate the same
+        new_events_total = np.round(size*old_events_total/old_detectable_events)
+        new_total_rate = self.rate_function(size, new_events_total, param_type=param_type, verbose=False)
+        meta_data["events_total"][-1] = new_events_total
+        meta_data["detectable_events"][-1] = size
+        meta_data["total_rate"][-1] = new_total_rate
+
+        print("collected number of detectable events = ", size)
+        print("total number of events = ", new_events_total)
+        print(f"total {param_type} event rate (yr^-1): {new_total_rate}")
+
+        # save the meta data
+        append_json(meta_data_path, meta_data, replace=True)
+        # save the final param dictionary
+        append_json(output_path, param_final, replace=True)
 
-        if trim_to_size:
-            # trim the final param dictionary
-            print(f"\n trmming final result to size={size}")
-            param_final = get_param_from_json(output_path)
-            # trim the final param dictionary
-            for key, value in param_final.items():
-                param_final[key] = param_final[key][:size]
+        return param_final, new_total_rate
 
-            # save the final param dictionary
-            append_json(output_path, param_final, replace=True)
+    def _append_meta_data(self, meta_data_path, n, events_total, total_rate):
+        """
+        Helper function for appending meta data json file.
 
-        return param_final
+        Parameters
+        ----------
+        meta_data_path : `str`
+            path to the metadata json file.
+        n : `int`
+            iterator.
+        events_total : `int`
+            total number of events.
+        total_rate : `float`
+            total rate (Mpc^-3 yr^-1).
+        """
 
+        # save meta data
+        meta_data = dict(events_total=[events_total], detectable_events=[float(n)], total_rate=[total_rate])
 
+        if os.path.exists(meta_data_path):
+            try:
+                append_json(meta_data_path, meta_data, replace=False)
+            except:
+                append_json(meta_data_path, meta_data, replace=True)
+        else:
+            append_json(meta_data_path, meta_data, replace=True)
 
+        print("collected number of detectable events = ", n)
+        print("total number of events = ", events_total)
+        print(f"total rate (yr^-1): {total_rate}")