edited test-code