Fix/Write Unit tests

[cyberphantom52]

The objective of this task is to review, update, and create unit tests for our project. Unit tests are essential for ensuring the correctness and robustness of our codebase, and they play a crucial role in maintaining code quality.