chore: Added unit tests