adding the test code