Added unit test function as separate module