Deduplicate repeated code

[Y0GAAAA]

Deduplicating repeated code by using a closure and calling it as needed.

[abhimanyu003]

Hello @Y0GAAAA thanks so much, for the contribution, I have merged your other PR.
This one is breaking a few test cases.
You can check here https://github.com/abhimanyu003/qubit/runs/4127862989?check_suite_focus=true#step:4:74

[abhimanyu003]

@harshvishu provided some cool updates, I have to close this PR as we can build up on that one.
Once thanks so much for your work and contribution.