finishing the test code