Adding a test for the function.