edited test code