Update the geometry optimization function

[ddhawan11]

Context:
Updated the geometry optimization function to perform a single task

Description of the Change:

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[github-actions]

Hello. You may have forgotten to update the changelog!
Please edit doc/releases/changelog-dev.md with:

• A one-to-two sentence description of the change. You may include a small working example for new features.
• A link back to this PR.
• Your name (or GitHub username) in the contributors section.

[soranjh]

Thanks @ddhawan11.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.66%. Comparing base (bbcf0fb) to head (a1f3d48).
Report is 1 commits behind head on master.

Additional details and impacted files

@@           Coverage Diff           @@
##           master    #6666   +/-   ##
=======================================
  Coverage   99.66%   99.66%           
=======================================
  Files         464      464           
  Lines       43961    43962    +1     
=======================================
+ Hits        43812    43813    +1     
  Misses        149      149           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[austingmhuang]

Make sure to address codecov before merging.