Can you add unit tests for this module?