Unit tests

[khattakdev]

Review and update the src/tests folder with test cases.

[Manas-Kenge]

@khattakdev Could you please elaborate more on this issue?

[khattakdev]

@Manas-Kenge Write unit tests using Jest for the code.