update function and add declaration

[AndrewZuo01]

🔍 What type of PR is this?

👀 What this PR does / why we need it:

• [ ] My pull request adheres to the code style of this project
• [ ] My code requires changes to the documentation
• [ ] I have updated the documentation as required
• [ ] All the tests have passed

🅰 Which issue(s) this PR fixes:

Fixes #1

📝 Special notes for your reviewer:

🎯 Describe how to verify it

📑 Additional documentation e.g., RFC, notion, Google docs, usage docs, etc.:

[pull-request-size[bot]]

Whoa! Easy there, Partner!

This PR is too big. Please break it up into smaller PRs.

[sweep-ai[bot]]

Apply Sweep Rules to your PR?

• [ ] Apply: All new business logic should have corresponding unit tests.
• [ ] Apply: Refactor large functions to be more modular.
• [ ] Apply: Add docstrings to all functions and file headers.

This is an automated message generated by Sweep AI.