add the test code