Little code Optimization

[Darknez07]

Description

Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes # (issue)

Type of change

Please delete options that are not relevant.

• [x] Optimized code

Checklist:

• [x] My code follows the style guidelines of this project
• [x] I have performed a self-review of my own code
• [x] I have commented my code, particularly in hard-to-understand areas
• [x] I have made corresponding changes to the documentation
• [x] My changes generate no new warnings
• [x] I have added tests that prove my fix is effective or that my feature works
• [x] Any dependent changes have been merged and published in downstream modules

[Darknez07]

@vinitshahdeo @Kashish121 @ishika1727

[Darknez07]

@Kashish121 would you like to take a look

[Darknez07]

@vinitshahdeo Is this fine or some more changes are required?