Yes, I will put the test code into a function.