Added the inline comments

[anirbansaha782]

Description

Please include a summary of the change, the motivation, and any additionl context that will help others understand your PR. If it closes one or more open issues, please tag them as described here.

Affected Dependencies

List any dependencies that are required for this change.

How has this been tested?

• Describe the tests that you ran to verify your changes.
• Provide instructions so we can reproduce.
• List any relevant details for your test configuration.

Checklist

• [ ] I have followed the Contribution Guidelines and Code of Conduct
• [ ] I have commented my code following the OpenMined Styleguide
• [ ] I have labeled this PR with the relevant Type labels
• [ ] My changes are covered by tests

[anirbansaha782]

Please review my pull request