made changes to test code