Mind removing the test cases? This file should just contain the function.