improving and fixing some code

[thevipuljha]

Description

Related Issue

This PR closes #

Motivation and Context

Screenshots (if any)

Types of changes

• [ ] Bug fix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [ ] Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

• [ ] My code follows the code style of this project.
• [ ] My change requires a change to the documentation.
• [ ] I have updated the documentation accordingly.
• [ ] I have read the CONTRIBUTING document.

[thevipuljha]

now

[SVijayB]

Hey @Vipul1142 thanks for the pull request! I've merged it 👍