Enhance test coverage #70
Description
It would be great to upgrade our test coverage in the project. Improving coverage can help us catch bugs earlier and ensure our code is more reliable. By adding more tests, especially in untested areas, can better handle edge cases and keep the application stable.