Mind removing the test cases. Let's just have the function.