made more changes to test code