updated the test with module func implemetation