Add test case for functions in modules