Enhance Test Coverage in Test Suite

To ensure the robustness and reliability of our application, I propose enhancing the test coverage in our test suite. This will help us identify potential issues early and maintain high code quality. Specifically, we should:

1. **Identify Critical Areas**: Focus on areas with the highest impact on functionality.
2. **Increase Unit Tests**: Add more unit tests to cover edge cases and different scenarios.
3. **Follow TDD**: Implement tests before adding any feature.